diff --git a/app/yolov5/train_server.py b/app/yolov5/train_server.py index b913f63..03b679c 100644 --- a/app/yolov5/train_server.py +++ b/app/yolov5/train_server.py @@ -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: