ckpt

2022-10-03 16:09:05 +05:30 · 2022-10-03 16:09:05 +05:30 · 1585911767
parent c8815fa969
commit 1585911767
1 changed files with 1 additions and 1 deletions
--- a/cli/train.py
+++ b/cli/train.py
@ -56,7 +56,7 @@ def main(config: DictConfig):
    trainer = instantiate(config.trainer,logger=logger,callbacks=callbacks)
    trainer.fit(model)
    if os.path.exists("./model/"):
-        logger.experiment.log_artifact(logger.run_id,f"./model/.*")
+        logger.experiment.log_artifact(logger.run_id,f"model_{JOB_ID}.ckpt")