Update cli.py

t5s/cli.py

@@ -44,6 +44,9 @@ parser_start.add_argument(
 )
 parser_start.add_argument(
     "-lr", "--learning-rate", default=0.0001, help="Enter the number of epochs"
+)
+parser_start.add_argument(
+    "-b", "--batch-size", default=2, help="Enter the number of batches"
 parser_dirs = command_subparser.add_parser(
     "dirs",
     help="Create directories that are ignored by git but required for " "the project",