From 9454e954e36f79d59b20109742fbdbeeba0d8475 Mon Sep 17 00:00:00 2001 From: Dick Ameln Date: Wed, 11 May 2022 14:17:59 +0200 Subject: [PATCH] add input_info to nncf config when not defined by user --- anomalib/config/config.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/anomalib/config/config.py b/anomalib/config/config.py index f35d3d312b..0dd8cfb046 100644 --- 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: