From 56b4d8b889da9d21083a7007f3b170252d5dd3b4 Mon Sep 17 00:00:00 2001
From: Felix <freehuang100@163.com>
Date: Mon, 31 May 2021 15:49:57 +0800
Subject: [PATCH] Update multilabel_dataset.py

---
 ppcls/data/dataset/multilabel_dataset.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/ppcls/data/dataset/multilabel_dataset.py b/ppcls/data/dataset/multilabel_dataset.py
index c1155500..2a797350 100644
--- a/ppcls/data/dataset/multilabel_dataset.py
+++ b/ppcls/data/dataset/multilabel_dataset.py
@@ -30,9 +30,10 @@ from ppcls.data import preprocess
 from ppcls.data.preprocess import transform
 from ppcls.utils import logger
 
+from .common_dataset import CommonDataset
 
 
-class MultiLabelDataset(Dataset):
+class MultiLabelDataset(CommonDataset):
 
     def _load_anno(self):
         assert os.path.exists(self._cls_path)
-- 
GitLab