Fixed output of time_remaining for multi-GPU

pull/5550/head
AlexeyAB 5 years ago
parent d4c10b0262
commit 08bc0c9373
  1. 2
      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;

Loading…
Cancel
Save