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
|