训练自己的数据集

[Q-amazing]

代码中说Dataset related classes and methods for ViT training and validation
Cifar10, Cifar100 and ImageNet2012 are supported。
我想问下，我想训练自己的数据集需要自己重新写datase类吗？

[xperzy]

是的，因为我们无法得知您数据集的具体信息和使用方法，所以需要您实现自己的 Dataset 类，然后在项目中（如果是分类任务）dataset.py的 get_dataset 方法中添加您的Dataset class。