This commit is contained in:
552068321@qq.com 2022-11-08 09:17:12 +08:00
parent e31203fafb
commit 0465da97a1

View File

@ -468,6 +468,7 @@ def train(hyp, opt, device, data_list,id,callbacks): # hyp is path/to/hyp.yaml
if RANK in {-1, 0}:
LOGGER.info(f'\n{epoch - start_epoch + 1} epochs completed in {(time.time() - t0) / 3600:.3f} hours.')
for f in best:
print('##################',f)
if os.path.exists(f):
strip_optimizer(f) # strip optimizers
if f is best: