From 4be71483bca87ffbc98d290b977eeddf3ba9c2b8 Mon Sep 17 00:00:00 2001 From: Songki Choi <songki.choi@intel.com> Date: Fri, 11 Nov 2022 15:48:32 +0900 Subject: [PATCH] Fixed merge error Signed-off-by: Songki Choi <songki.choi@intel.com> --- external/model-preparation-algorithm/mpa_tasks/apis/task.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/external/model-preparation-algorithm/mpa_tasks/apis/task.py b/external/model-preparation-algorithm/mpa_tasks/apis/task.py index 2063e628d48..1ffeb18088f 100644 --- a/external/model-preparation-algorithm/mpa_tasks/apis/task.py +++ b/external/model-preparation-algorithm/mpa_tasks/apis/task.py @@ -90,9 +90,6 @@ def _run_task(self, stage_module, mode=None, dataset=None, parameters=None, **kw self._model_cfg["model_classes"] = model_classes if dataset is not None: train_data_cfg = Stage.get_data_cfg(self._data_cfg, "train") - # if dataset size is smaller than batch size - if 0 < len(dataset) < self._recipe_cfg.data.get("samples_per_gpu", 2): - train_data_cfg.drop_last = False train_data_cfg["data_classes"] = data_classes new_classes = np.setdiff1d(data_classes, model_classes).tolist() train_data_cfg["new_classes"] = new_classes