Releases: Kedreamix/Pytorch-Image-Classification
Releases · Kedreamix/Pytorch-Image-Classification
MAE in CIFAR10
猫狗数据集(用于迁移学习)
迁移学习Transfer Learning的小型猫狗数据集,训练集2000张,验证集1000张,测试集1000张
除此之外,加上在整个猫狗数据集上训练后的最优权重
也会把完整的kaggle的猫狗数据集放上来,有兴趣可以自己下载,不过我们迁移学习用的是小型的数据集,完整数据集中,训练集有2w张,验证集各5000张,测试集1w张