diff --git a/train.py b/train.py index f1d3f757..1473b939 100644 --- a/train.py +++ b/train.py @@ -330,7 +330,7 @@ def train(): # Report time plot_results() # save as results.png - print('%g epochs completed in %.3f hours.' % (epoch - start_epoch + 1, (time.time() - t0) / 3600)) + print('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600)) dist.destroy_process_group() if torch.cuda.device_count() > 1 else None torch.cuda.empty_cache() return results diff --git a/utils/utils.py b/utils/utils.py index 041c602f..94858c30 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -555,7 +555,7 @@ def get_yolo_layers(model): def print_model_biases(model): # prints the bias neurons preceding each yolo layer - print('\nModel Output-Bias Summary::') + print('\nModel output-bias Summary:') for l in model.yolo_layers: # print pretrained biases b = model.module_list[l - 1][0].bias.view(3, -1) # bias 3x85 print('regression: %.2f+/-%.2f, ' % (b[:, :4].mean(), b[:, :4].std()),