From af61da5d41039470721bb918fcb92c364e73966c Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Tue, 2 Apr 2019 14:07:14 +0200 Subject: [PATCH] updates --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train.py b/train.py index 58196e65..589a9f7b 100644 --- a/train.py +++ b/train.py @@ -185,8 +185,8 @@ def train( if best_loss == mloss['total']: os.system('cp ' + latest + ' ' + best) - # Save backup weights every 5 epochs (optional) - if epoch > 0 and epoch % 5 == 0: + # Save backup weights every 10 epochs (optional) + if epoch > 0 and epoch % 10 == 0: os.system('cp ' + latest + ' ' + weights + 'backup%g.pt' % epoch) # Calculate mAP