zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge remote-tracking branch 'origin/master'

This commit is contained in:
Glenn Jocher 2020-06-30 16:42:16 -07:00
parent 4e62eb90be 5c1c73ecda
commit 0a08375a8a
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
train.py
View File

@ -133,9 +133,13 @@ def train(hyp):
with open(results_file, 'w') as file: with open(results_file, 'w') as file:
file.write(ckpt['training_results']) # write results.txt file.write(ckpt['training_results']) # write results.txt
# epochs
start_epoch = ckpt['epoch'] + 1 start_epoch = ckpt['epoch'] + 1
assert opt.epochs > start_epoch, '%s has already trained %g epochs. --epochs must be greater than %g' % \ if epochs < start_epoch:
(opt.weights, ckpt['epoch'], ckpt['epoch']) print('%s has been trained for %g epochs. Fine-tuning for %g additional epochs.' %
(opt.weights, ckpt['epoch'], epochs))
epochs += ckpt['epoch'] # finetune additional epochs
del ckpt del ckpt
# Mixed precision training https://github.com/NVIDIA/apex # Mixed precision training https://github.com/NVIDIA/apex