This commit is contained in:
iperov 2021-07-17 22:03:59 +04:00
parent 0748b8d043
commit 8b71b83c94

View File

@ -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()