gagan3012 commited on
Commit
2fca2c2
·
1 Parent(s): ab59105

Update cli.py

Browse files
Files changed (1) hide show
  1. t5s/cli.py +2 -0
t5s/cli.py CHANGED
@@ -134,6 +134,8 @@ def start(arguments):
134
  with open("model_params.yml") as f:
135
  newdct = yaml.safe_load(f)
136
  newdct.update(model_params)
 
 
137
  def parse_args(args):
138
  arguments = vars(arg_parser.parse_args(args=args or ["--help"]))
139
  return arguments
 
134
  with open("model_params.yml") as f:
135
  newdct = yaml.safe_load(f)
136
  newdct.update(model_params)
137
+ with open("model_params.yml", "w") as f:
138
+ yaml.dump(newdct, f)
139
  def parse_args(args):
140
  arguments = vars(arg_parser.parse_args(args=args or ["--help"]))
141
  return arguments