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

This commit is contained in:
Siddhant Sadangi 2023-06-08 19:16:51 +02:00 committed by GitHub
parent e88c89d269
commit 29c954a138
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

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