Remove Neptune `run.stop()` in `on_train_end()` (#3103)

2 years ago · 29c954a138
parent e88c89d269
commit 29c954a138
1 changed files with 0 additions and 1 deletions
--- 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 = {