From 158591176708c817bdaec3a9180fbe4f06564c0a Mon Sep 17 00:00:00 2001
From: shahules786 <shahules786@gmail.com>
Date: Mon, 3 Oct 2022 16:09:05 +0530
Subject: [PATCH] ckpt

---
 cli/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cli/train.py b/cli/train.py
index 8a4beaf..cd19ffc 100644
--- 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")