Commit 2303d51e authored by Federico Rossi's avatar Federico Rossi

...

parent 7839754b
......@@ -10,7 +10,7 @@ using namespace tiny_dnn;
int main(int argc, char **argv) {
tiny_dnn::models::alexnet<227,227,1000> net;
for(int i = 0; i < 50; ++i) {
for(int i = 0; i < 1; ++i) {
std::cout << i << "/50" << std::endl;
net.predict(image<tiny_dnn::float_t>(shape3d(227,227,3),image_type::rgb));
}
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment