From d1601ae0f31de5a8b0a273a6a1d04ca172e7acd8 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Wed, 8 Apr 2020 21:34:34 -0700 Subject: [PATCH] training updates --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 6c204b7f..5c470e3e 100644 --- a/train.py +++ b/train.py @@ -209,7 +209,7 @@ def train(): # Start training nb = len(dataloader) # number of batches - n_burn = max(3 * nb, 500) # burn-in iterations, max(3 epochs, 300 iterations) + n_burn = max(3 * nb, 500) # burn-in iterations, max(3 epochs, 500 iterations) maps = np.zeros(nc) # mAP per class # torch.autograd.set_detect_anomaly(True) results = (0, 0, 0, 0, 0, 0, 0) # 'P', 'R', 'mAP', 'F1', 'val GIoU', 'val Objectness', 'val Classification'