diff --git a/ultralytics/yolo/utils/callbacks/neptune.py b/ultralytics/yolo/utils/callbacks/neptune.py index 96cb049..be64341 100644 --- a/ultralytics/yolo/utils/callbacks/neptune.py +++ b/ultralytics/yolo/utils/callbacks/neptune.py @@ -93,7 +93,6 @@ def on_train_end(trainer): # Log the final model run[f'weights/{trainer.args.name or trainer.args.task}/{str(trainer.best.name)}'].upload(File(str( trainer.best))) - run.stop() callbacks = {