src/classifier.c
@@ -3,6 +3,7 @@ #include "parser.h" #include "option_list.h" #include "blas.h" #include "classifier.h" #include <sys/time.h> #ifdef OPENCV @@ -49,7 +50,7 @@ load_weights(&net, weightfile); } printf("Learning Rate: %g, Momentum: %g, Decay: %g\n", net.learning_rate, net.momentum, net.decay); int imgs = 1024; int imgs = net.batch; list *options = read_data_cfg(datacfg); @@ -72,7 +73,7 @@ args.w = net.w; args.h = net.h; args.min = net.w; args.min = net.min_crop; args.max = net.max_crop; args.size = net.w;