search

kanade00 / Federated_Anomaly_detection

4 stars 1 forks source link

FileNotFoundError: Couldn't find any class folder in ./data/pseudo_imagenet. #2

Open tianhe042 opened 4 months ago

tianhe042 commented 4 months ago

您好，在下载完cifar10和cifar100之后提示如题报错，请问一下这个pseudo_imagenet的数据集在哪里可以下载呢？

tianhe042 commented 4 months ago

运行指令为CUDA_VISIBLE_DEVICES=1 python federated_train_sed.py --arch resnet18 --training-mode SimCLR --shift_trans_type rotation --clusters 10 --ratio_pollution 0.01

kanade00 commented 4 months ago

pseudo_imagenet是我用biggan模型生成的类似ImgaeNet数据的数据集，具体可以参考我上传的generate_data.py文件，需要你下载生成模型的ckpt