Skip to content

Commit 861157a

Browse files
authored
Fix default dict (#2025)
fix: change default to configdict Signed-off-by: Inhyuk Andy Cho <andy.inhyuk.jo@intel.com>
1 parent 4000c7d commit 861157a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

otx/algorithms/common/tasks/nncf_task.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -162,7 +162,7 @@ def _prepare_optimize(self):
162162
# last batch size of 1 causes undefined behaviour for batch normalization
163163
# when initializing and training NNCF
164164
if self._data_cfg is not None:
165-
data_loader = self._recipe_cfg.data.get("train_dataloader", {})
165+
data_loader = self._recipe_cfg.data.get("train_dataloader", ConfigDict())
166166
samples_per_gpu = data_loader.get("samples_per_gpu", self._recipe_cfg.data.get("samples_per_gpu"))
167167
otx_dataset = get_configs_by_keys(self._data_cfg.data.train, "otx_dataset")
168168
assert len(otx_dataset) == 1

0 commit comments

Comments
 (0)