Anuj-Panthri commited on
Commit
a36040c
·
1 Parent(s): fd0734b

changed model name while logging to comet

Browse files
Files changed (1) hide show
  1. src/scripts/train.py +1 -1
src/scripts/train.py CHANGED
@@ -57,7 +57,7 @@ def train(args):
57
  # log model to comet
58
  if "LOCAL_SYSTEM" not in os.environ:
59
  if experiment:
60
- experiment.log_model(f"{config.task}_{config.dataset}_{config.model}",model_dir)
61
 
62
  # evaluate model
63
  print_title("\nEvaluating Model")
 
57
  # log model to comet
58
  if "LOCAL_SYSTEM" not in os.environ:
59
  if experiment:
60
+ experiment.log_model(f"model",model_dir)
61
 
62
  # evaluate model
63
  print_title("\nEvaluating Model")