|
|
@ -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)) { |
|
|
|