Spaces:
Runtime error
Runtime error
Update cli.py
Browse files- t5s/cli.py +1 -0
t5s/cli.py
CHANGED
@@ -126,6 +126,7 @@ def start(arguments):
|
|
126 |
"model_type": arguments["model_type"],
|
127 |
"model_name": arguments["model_name"],
|
128 |
"epochs": arguments["epochs"],
|
|
|
129 |
def parse_args(args):
|
130 |
arguments = vars(arg_parser.parse_args(args=args or ["--help"]))
|
131 |
return arguments
|
|
|
126 |
"model_type": arguments["model_type"],
|
127 |
"model_name": arguments["model_name"],
|
128 |
"epochs": arguments["epochs"],
|
129 |
+
"learning_rate": arguments["learning_rate"],
|
130 |
def parse_args(args):
|
131 |
arguments = vars(arg_parser.parse_args(args=args or ["--help"]))
|
132 |
return arguments
|