From d66010956d201a44204d49e9dad8e7e83fb5bf79 Mon Sep 17 00:00:00 2001 From: zengshao0622 Date: Tue, 14 Mar 2023 07:21:55 +0000 Subject: [PATCH] fix cae dataset --- ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml | 8 ++++---- ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml b/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml index 99447275..70ac8c30 100644 --- a/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml +++ b/ppcls/configs/CAE/cae_base_patch16_224_finetune.yaml @@ -69,8 +69,8 @@ DataLoader: Train: dataset: name: ImageNetDataset - image_root: ./dataset/flowers102/ - cls_label_path: ./dataset/flowers102/train_list.txt + image_root: ./dataset/ILSVRC2012/ + cls_label_path: ./dataset/ILSVRC2012/train_list.txt batch_transform_ops: - MixupCutmixHybrid: mixup_alpha: 0.8 @@ -114,8 +114,8 @@ DataLoader: Eval: dataset: name: ImageNetDataset - image_root: ./dataset/flowers102/ - cls_label_path: ./dataset/flowers102/val_list.txt + image_root: /dataset/ILSVRC2012/ + cls_label_path: ./dataset/ILSVRC2012/val_list.txt transform_ops: - DecodeImage: to_rgb: True diff --git a/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml b/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml index 579163d9..a8a34691 100644 --- a/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml +++ b/ppcls/configs/CAE/cae_large_patch16_224_finetune.yaml @@ -69,8 +69,8 @@ DataLoader: Train: dataset: name: ImageNetDataset - image_root: ./dataset/flowers102/ - cls_label_path: ./dataset/flowers102/train_list.txt + image_root: ./dataset/ILSVRC2012/ + cls_label_path: ./dataset/ILSVRC2012/train_list.txt batch_transform_ops: - MixupCutmixHybrid: mixup_alpha: 0.8 @@ -114,8 +114,8 @@ DataLoader: Eval: dataset: name: ImageNetDataset - image_root: ./dataset/flowers102/ - cls_label_path: ./dataset/flowers102/val_list.txt + image_root: ./dataset/ILSVRC2012/ + cls_label_path: ./dataset/ILSVRC2012/val_list.txt transform_ops: - DecodeImage: to_rgb: True -- GitLab