fix `batch`

1c62da1d · baberabb · 38e3b73f · 1c62da1d
Commit 1c62da1d authored Dec 06, 2023 by baberabb
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

scripts/model_comparator.py scripts/model_comparator.py +1 -1

No files found.
--- a/scripts/model_comparator.py
+++ b/scripts/model_comparator.py
@@ -77,7 +77,7 @@ def parse_args():
    )
    parser.add_argument(
        "--batch",
-        type=int,
+        type=str,
        default=8,
    )
    parser.add_argument(