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

2023-06-08 19:16:51 +02:00 · 2023-06-08 19:16:51 +02:00 · 29c954a138
commit 29c954a138
parent e88c89d269
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 = {