added option for strict model loading

cvfinetune/finetuner/base.py

@@ -148,7 +148,12 @@ class _ModelMixin(abc.ABC):
 				loader_func = self.model.load_for_finetune
 
 			logging.info(msg.format(self.weights))
-			loader = partial(loader_func, weights=self.weights, headless=args.headless)
+			kwargs = dict(
+				weights=self.weights,
+				strict=args.load_strict,
+				headless=args.headless,
+			)
+			loader = partial(loader_func, **kwargs)
 
 		feat_size = self.model.meta.feature_size
 

cvfinetune/parser/model_args.py

@@ -40,8 +40,8 @@ def add_model_args(parser):
 		Arg("--headless", action="store_true",
 			help="ignores classifier layer during loading"),
 
-		Arg("--force_load", action="store_true",
-			help="force loading from caffe model"),
+		Arg("--load_strict", action="store_true",
+			help="load weights in a strict mode"),
 	]
 
 	parser.add_args(_args, group_name="Model arguments")