Spaces:
Runtime error
Runtime error
Update cli.py
Browse files- 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
|