add dataset alias for PaddleX (#2893)

83adff2a · wuyefeilin · GitHub · 607a07cb · 83adff2a
隐藏空白更改
内联并排

Showing with 5 addition and 0 deletion

ppcls/data/__init__.py ppcls/data/__init__.py +5 -0

未找到文件。
--- 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