From 29c954a1385815e522f0d94d7144814cc4e6da42 Mon Sep 17 00:00:00 2001
From: Siddhant Sadangi <siddhant.sadangi@gmail.com>
Date: Thu, 8 Jun 2023 19:16:51 +0200
Subject: [PATCH] Remove Neptune `run.stop()` in `on_train_end()` (#3103)

---
 ultralytics/yolo/utils/callbacks/neptune.py | 1 -
 1 file changed, 1 deletion(-)

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 = {