We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 4000c7d commit 861157aCopy full SHA for 861157a
otx/algorithms/common/tasks/nncf_task.py
@@ -162,7 +162,7 @@ def _prepare_optimize(self):
162
# last batch size of 1 causes undefined behaviour for batch normalization
163
# when initializing and training NNCF
164
if self._data_cfg is not None:
165
- data_loader = self._recipe_cfg.data.get("train_dataloader", {})
+ data_loader = self._recipe_cfg.data.get("train_dataloader", ConfigDict())
166
samples_per_gpu = data_loader.get("samples_per_gpu", self._recipe_cfg.data.get("samples_per_gpu"))
167
otx_dataset = get_configs_by_keys(self._data_cfg.data.train, "otx_dataset")
168
assert len(otx_dataset) == 1
0 commit comments