From 83adff2acc5a0defb5457287febeec1ca02640ec Mon Sep 17 00:00:00 2001
From: wuyefeilin <30919197+wuyefeilin@users.noreply.github.com>
Date: Wed, 2 Aug 2023 10:41:25 +0800
Subject: [PATCH] add dataset alias for PaddleX (#2893)

---
 ppcls/data/__init__.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/ppcls/data/__init__.py b/ppcls/data/__init__.py
index bc61139f..4cba5532 100644
--- a/ppcls/data/__init__.py
+++ b/ppcls/data/__init__.py
@@ -189,3 +189,8 @@ def build_dataloader(config, mode, device, use_dali=False, seed=None):
 
     logger.debug("build data_loader({}) success...".format(data_loader))
     return data_loader
+
+
+# for PaddleX
+ClsDataset = ImageNetDataset
+ShiTuDataset = ImageNetDataset
-- 
GitLab