eubinecto commited on
Commit
2b6388f
1 Parent(s): 64a6414

[#1] the name of the artifact should just be seq2seq

Browse files
Files changed (1) hide show
  1. main_train.py +1 -1
main_train.py CHANGED
@@ -46,7 +46,7 @@ def main():
46
  if not config['fast_dev_run'] and trainer.current_epoch == config['max_epochs'] - 1:
47
  ckpt_path = ROOT_DIR / "model.ckpt"
48
  trainer.save_checkpoint(str(ckpt_path))
49
- artifact = wandb.Artifact(name=config['model'], type="model", metadata=config)
50
  artifact.add_file(str(ckpt_path))
51
  run.log_artifact(artifact, aliases=["latest", config['ver']])
52
  os.remove(str(ckpt_path)) # make sure you remove it after you are done with uploading it
 
46
  if not config['fast_dev_run'] and trainer.current_epoch == config['max_epochs'] - 1:
47
  ckpt_path = ROOT_DIR / "model.ckpt"
48
  trainer.save_checkpoint(str(ckpt_path))
49
+ artifact = wandb.Artifact(name="seq2seq", type="model", metadata=config)
50
  artifact.add_file(str(ckpt_path))
51
  run.log_artifact(artifact, aliases=["latest", config['ver']])
52
  os.remove(str(ckpt_path)) # make sure you remove it after you are done with uploading it