Skip to content
Snippets Groups Projects

Batch predictions

Merged Yoann Schneider requested to merge batch-pred into main
5 files
+ 282
83
Compare changes
  • Side-by-side
  • Inline
Files
5
+ 8
0
@@ -140,4 +140,12 @@ def add_predict_parser(subcommands) -> None:
type=int,
required=False,
)
parser.add_argument(
"--batch-size",
help="Size of prediction batches.",
type=int,
default=1,
required=False,
)
parser.set_defaults(func=run)
Loading