➕ add input_info to nncf config when not defined by user (#307)

openvinotoolkit · May 12, 2022 · 9b117b0 · 9b117b0
1 parent 995aa8c
commit 9b117b0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/anomalib/config/config.py b/anomalib/config/config.py
@@ -64,6 +64,8 @@ def update_nncf_config(config: Union[DictConfig, ListConfig]) -> Union[DictConfi
     sample_size = (crop_size, crop_size) if isinstance(crop_size, int) else crop_size
     if "optimization" in config.keys():
         if "nncf" in config.optimization.keys():
+            if "input_info" not in config.optimization.nncf.keys():
+                config.optimization.nncf["input_info"] = {"sample_size": None}
             config.optimization.nncf.input_info.sample_size = [1, 3, *sample_size]
             if config.optimization.nncf.apply:
                 if "update_config" in config.optimization.nncf: