diff --git a/train.py b/train.py index a5fd778c..f1d3f757 100644 --- a/train.py +++ b/train.py @@ -268,11 +268,11 @@ def train(): '%g/%g' % (epoch, epochs - 1), '%.3gG' % mem, *mloss, len(targets), img_size) pbar.set_description(s) # end batch ----------------------------------------------------------------------- + final_epoch = epoch + 1 == epochs if opt.prebias: print_model_biases(model) else: # Calculate mAP (always test final epoch, skip first 10 if opt.nosave) - final_epoch = epoch + 1 == epochs if not (opt.notest or (opt.nosave and epoch < 10)) or final_epoch: with torch.no_grad(): results, maps = test.test(cfg, diff --git a/utils/utils.py b/utils/utils.py index f13ccea1..041c602f 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -555,6 +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::') 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()),