Glenn Jocher
GitHub
3 years ago
1 changed files with 1 additions and 1 deletions
+ 1
- 1
train.py
View File
| # Test best.pt | # Test best.pt | ||||
| logger.info('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600)) | logger.info('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600)) | ||||
| if opt.data.endswith('coco.yaml') and nc == 80: # if COCO | if opt.data.endswith('coco.yaml') and nc == 80: # if COCO | ||||
| for m in (last, best) if best.exists() else (last): # speed, mAP tests | |||||
| for m in [last, best] if best.exists() else [last]: # speed, mAP tests | |||||
| results, _, _ = test.test(opt.data, | results, _, _ = test.test(opt.data, | ||||
| batch_size=batch_size * 2, | batch_size=batch_size * 2, | ||||
| imgsz=imgsz_test, | imgsz=imgsz_test, |
Loading…