Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

ckpt

This commit is contained in:
shahules786 2022-10-03 16:09:05 +05:30
parent c8815fa969
commit 1585911767
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
cli/train.py
View File

@ -56,7 +56,7 @@ def main(config: DictConfig):
trainer = instantiate(config.trainer,logger=logger,callbacks=callbacks) trainer = instantiate(config.trainer,logger=logger,callbacks=callbacks)
trainer.fit(model) trainer.fit(model)
if os.path.exists("./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")