|
@@ -30,7 +30,7 @@ def add_model_args(parser: BaseParser, *,
|
|
|
Arg("--input_size", type=int, nargs="+", default=0,
|
|
Arg("--input_size", type=int, nargs="+", default=0,
|
|
|
help="overrides default input size of the model, if greater than 0"),
|
|
help="overrides default input size of the model, if greater than 0"),
|
|
|
|
|
|
|
|
- Arg("--parts_input_size", type=int, nargs="+", default=0,
|
|
|
|
|
|
|
+ Arg("--part_input_size", type=int, nargs="+", default=0,
|
|
|
help="overrides default input part size of the model, if greater than 0"),
|
|
help="overrides default input part size of the model, if greater than 0"),
|
|
|
|
|
|
|
|
PrepareType.as_arg("prepare_type",
|
|
PrepareType.as_arg("prepare_type",
|
