Glenn Jocher
4 years ago
1 changed files with 2 additions and 0 deletions
+ 2
- 0
train.py
View File
| file.write(ckpt['training_results']) # write results.txt | file.write(ckpt['training_results']) # write results.txt | ||||
| 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' % \ | |||||
| (opt.weights, ckpt['epoch'], ckpt['epoch']) | |||||
| del ckpt | del ckpt | ||||
| # Mixed precision training https://github.com/NVIDIA/apex | # Mixed precision training https://github.com/NVIDIA/apex |
Loading…