Merge remote-tracking branch 'origin/master'

2020-06-30 15:25:45 -07:00 · 2020-06-30 15:25:45 -07:00 · 2b7a11300e
parent a9dc0c2c29 b9da3ea46a
commit 2b7a11300e
1 changed files with 2 additions and 0 deletions
--- a/train.py
+++ b/train.py
@ -134,6 +134,8 @@ def train(hyp):
                file.write(ckpt['training_results'])  # write results.txt

        start_epoch = ckpt['epoch'] + 1
+        assert opt.epochs > start_epoch, '%s has already trained %g epochs. --epochs must be greater than %g' % \
+                                         (opt.weights, ckpt['epoch'], ckpt['epoch'])
        del ckpt

    # Mixed precision training https://github.com/NVIDIA/apex