From 8b71b83c940ef63414a9d88780521acba1760974 Mon Sep 17 00:00:00 2001 From: iperov Date: Sat, 17 Jul 2021 22:03:59 +0400 Subject: [PATCH] fix --- mainscripts/Merger.py | 1 + 1 file changed, 1 insertion(+) diff --git a/mainscripts/Merger.py b/mainscripts/Merger.py index fba37f1..0703dc1 100644 --- a/mainscripts/Merger.py +++ b/mainscripts/Merger.py @@ -49,6 +49,7 @@ def main (model_class_name=None, model = models.import_model(model_class_name)(is_training=False, saved_models_path=saved_models_path, force_gpu_idxs=force_gpu_idxs, + force_model_name=force_model_name, cpu_only=cpu_only) predictor_func, predictor_input_shape, cfg = model.get_MergerConfig()