Adds support to print remaining training time on console if darknet is not compiled with OpenCV

This commit is contained in:
Muhammad Maaz 2020-03-22 20:40:26 +05:00
parent bd7690b019
commit 8800212faa
1 changed files with 1 additions and 1 deletions

View File

@ -292,7 +292,7 @@ void train_detector(char *datacfg, char *cfgfile, char *weightfile, int *gpus, i
if (iteration < net.burn_in * 3) fprintf(stderr, "\n Tensor Cores are disabled until the first %d iterations are reached.", 3 * net.burn_in); if (iteration < net.burn_in * 3) fprintf(stderr, "\n Tensor Cores are disabled until the first %d iterations are reached.", 3 * net.burn_in);
else fprintf(stderr, "\n Tensor Cores are used."); else fprintf(stderr, "\n Tensor Cores are used.");
} }
printf("\n %d: %f, %f avg loss, %f rate, %lf seconds, %d images\n", iteration, loss, avg_loss, get_current_rate(net), (what_time_is_it_now() - time), iteration*imgs); printf("\n %d: %f, %f avg loss, %f rate, %lf seconds, %d images, %f time left\n", iteration, loss, avg_loss, get_current_rate(net), (what_time_is_it_now() - time), iteration*imgs, avg_time);
int draw_precision = 0; int draw_precision = 0;
if (calc_map && (iteration >= next_map_calc || iteration == net.max_batches)) { if (calc_map && (iteration >= next_map_calc || iteration == net.max_batches)) {