gagan3012 commited on
Commit
346a532
·
1 Parent(s): 2ec127a

Update cli.py

Browse files
Files changed (1) hide show
  1. t5s/cli.py +2 -0
t5s/cli.py CHANGED
@@ -130,6 +130,8 @@ def start(arguments):
130
  "batch_size": arguments["batch_size"],
131
  }
132
  with open("data_params.yml", "w") as f:
 
 
133
  def parse_args(args):
134
  arguments = vars(arg_parser.parse_args(args=args or ["--help"]))
135
  return arguments
 
130
  "batch_size": arguments["batch_size"],
131
  }
132
  with open("data_params.yml", "w") as f:
133
+ yaml.dump(data_params, f)
134
+ with open("model_params.yml") as f:
135
  def parse_args(args):
136
  arguments = vars(arg_parser.parse_args(args=args or ["--help"]))
137
  return arguments