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