diff --git a/src/detector.c b/src/detector.c index 8ec3f1ae..5d284e02 100644 --- a/src/detector.c +++ b/src/detector.c @@ -301,7 +301,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i int calc_map_for_each = 4 * train_images_num / (net.batch * net.subdivisions); // calculate mAP for each 4 Epochs calc_map_for_each = fmax(calc_map_for_each, 100); int next_map_calc = iter_map + calc_map_for_each; - next_map_calc = 1000;// fmax(next_map_calc, net.burn_in); + next_map_calc = fmax(next_map_calc, net.burn_in); //next_map_calc = fmax(next_map_calc, 400); if (calc_map) { printf("\n (next mAP calculation at %d iterations) ", next_map_calc);