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

5 years ago · 8800212faa
parent bd7690b019
commit 8800212faa
1 changed files with 1 additions and 1 deletions
--- a/src/detector.c
+++ b/src/detector.c
@ -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);
            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;
        if (calc_map && (iteration >= next_map_calc || iteration == net.max_batches)) {