train T2T-ViT on the custom dataset

[123-xin]

Hello, thank you very much for your work. Can you tell me how my training set and label format should be set up if I train the network on a custom dataset. (My custom dataset is divided into two categories )

[yuanli2333]

Hi, you should write a dataloader to load your own data,

Or if you want to use our repo to train, you should reorganize the data structure of your dataset as ImageNet:

│imagenet/ ├──train/ │ ├── n01440764 │ │ ├── n01440764_10026.JPEG │ │ ├── n01440764_10027.JPEG │ │ ├── ...... │ ├── ...... ├──val/ │ ├── n01440764 │ │ ├── ILSVRC2012_val_00000293.JPEG │ │ ├── ILSVRC2012_val_00002138.JPEG │ │ ├── ...... │ ├── ......