mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-29 02:14:22 +08:00
Remove Neptune run.stop()
in on_train_end()
(#3103)
This commit is contained in:
parent
e88c89d269
commit
29c954a138
@ -93,7 +93,6 @@ def on_train_end(trainer):
|
|||||||
# Log the final model
|
# Log the final model
|
||||||
run[f'weights/{trainer.args.name or trainer.args.task}/{str(trainer.best.name)}'].upload(File(str(
|
run[f'weights/{trainer.args.name or trainer.args.task}/{str(trainer.best.name)}'].upload(File(str(
|
||||||
trainer.best)))
|
trainer.best)))
|
||||||
run.stop()
|
|
||||||
|
|
||||||
|
|
||||||
callbacks = {
|
callbacks = {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user