fixed some part input size issues

cvfinetune/finetuner/base.py

@@ -223,7 +223,7 @@ class _DatasetMixin(abc.ABC):
 	def init_datasets(self, opts):
 
 		size = Size(opts.input_size)
-		part_size = getattr(opts, "part_size")
+		part_size = getattr(opts, "parts_input_size", None)
 		part_size = size if part_size is None else Size(part_size)
 
 		self.prepare = partial(PrepareType[opts.prepare_type](self.model),

cvfinetune/parser/model_args.py

@@ -25,6 +25,9 @@ def add_model_args(parser):
 		Arg("--input_size", type=int, nargs="+", default=0,
 			help="overrides default input size of the model, if greater than 0"),
 
+		Arg("--parts_input_size", type=int, nargs="+", default=0,
+			help="overrides default input part size of the model, if greater than 0"),
+
 		PrepareType.as_arg("prepare_type",
 			help_text="type of image preprocessing"),