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 ------------------------------------------------------------------