fix

2021-07-17 22:03:59 +04:00
parent 0748b8d043
commit 8b71b83c94
1 changed files with 1 additions and 0 deletions
@@ -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()