batch_size arg accepts str

f4df3e48 · Benjamin Fattori · 6a6a0ebb · f4df3e48
Commit f4df3e48 authored Jul 22, 2023 by Benjamin Fattori
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

main.py main.py +1 -1

No files found.
--- a/main.py
+++ b/main.py
@@ -31,7 +31,7 @@ def parse_args():
        default=0,
        help="Number of examples in few-shot context",
    )
-    parser.add_argument("--batch_size", type=int, default=1)  # TODO: only integers
+    parser.add_argument("--batch_size", type=str, default=1) 
    parser.add_argument(
        "--max_batch_size",
        type=int,