gagan3012 commited on
Commit
b82f5d2
·
1 Parent(s): 067d38d

Update cli.py

Browse files
Files changed (1) hide show
  1. 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