From 25a579e41755b048ce1a9758ed78bdd6e3936a1e Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 24 Aug 2019 17:16:20 +0200 Subject: [PATCH] removed xy/wh loss reporting --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 279a2fdd..3eed7edd 100644 --- a/train.py +++ b/train.py @@ -178,7 +178,7 @@ def train(): model_info(model, report='summary') # 'full' or 'summary' nb = len(dataloader) maps = np.zeros(nc) # mAP per class - results = (0, 0, 0, 0, 0, 0, 0, 0) # 'P', 'R', 'mAP', 'F1', 'val GIoU', 'val Objectness', 'val Classification' + results = (0, 0, 0, 0, 0, 0, 0) # 'P', 'R', 'mAP', 'F1', 'val GIoU', 'val Objectness', 'val Classification' t0 = time.time() for epoch in range(start_epoch, epochs): # epoch ------------------------------------------------------------------