From 60bc2c1fbddb74755a256ba6a9b18752660b5297 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 8 Jul 2019 15:43:46 +0200 Subject: [PATCH] updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index f980d69b..c9aadb70 100644 --- a/train.py +++ b/train.py @@ -155,7 +155,7 @@ def train( for epoch in range(start_epoch, epochs): model.train() print(('\n%8s%12s' + '%10s' * 7) % - ('Epoch', 'Batch', 'xy', 'wh', 'obj', 'cls', 'total', 'targets', 'img_size')) + ('Epoch', 'Batch', 'GIoU/xy', 'wh', 'obj', 'cls', 'total', 'targets', 'img_size')) # Update scheduler scheduler.step()