diff --git a/src/detector.c b/src/detector.c index cf26107d..d055d1ec 100644 --- a/src/detector.c +++ b/src/detector.c @@ -348,7 +348,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i draw_precision = 1; } - time_remaining = (net.max_batches - iteration)*(what_time_is_it_now() - time + load_time) / 60 / 60; + time_remaining = ((net.max_batches - iteration) / ngpus)*(what_time_is_it_now() - time + load_time) / 60 / 60; // set initial value, even if resume training from 10000 iteration if (avg_time < 0) avg_time = time_remaining; else avg_time = alpha_time * time_remaining + (1 - alpha_time) * avg_time;