Remove Neptune run.stop()
in on_train_end()
(#3103)
This commit is contained in:
@ -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 = {
|
||||||
|
Reference in New Issue
Block a user