Adds --pad_max flag and relevant docs.

It is not plugged into anything yet. Working on issue CUNY-CL#50.
kylebgorman · May 29, 2023 · 4543c5b · 4543c5b
1 parent 6023dbc
commit 4543c5b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -153,6 +153,7 @@ A non-exhaustive list includes:
 
 -   Batch size:
     -   `--batch_size` (default: `32`)
+    -   `--pad_max` (default: not enabled): gives a consistent batch size
 -   Regularization:
     -   `--dropout` (default: `.2`)
     -   `--label_smoothing` (default: not enabled)

diff --git a/yoyodyne/train.py b/yoyodyne/train.py
@@ -413,6 +413,12 @@ def main() -> None:
         default=defaults.BATCH_SIZE,
         help="Batch size. Default: %(default)s.",
     )
+    parser.add_argument(
+        "--pad_max",
+        action="store_true",
+        help="Pads all batches to the same length. Default: False.",
+    )
+    parser.add_argument("--no_pad_max", action="store_false", dest="pad_max")
     parser.add_argument(
         "--patience", type=int, help="Patience for early stopping"
     )