assert --epochs
This commit is contained in:
1 parent
615d6d0cfa
commit
3b16c865f0
1 file changed
+2
@@ -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
|
||||
|
||||
Reference in new issue
Block a user