search

microsoft / esvit

EsViT: Efficient self-supervised Vision Transformers
MIT License
408 stars 46 forks source link

Training on custom dataset #2

Closed madr3z closed 3 years ago

madr3z commented 3 years ago

What a custom dataset structure should be like and how to train on it? Let's say I have a dataset of two classes as the folder (binary): 1. Has cat, 2. No cat. In each sub-folder, there are images. What changes to the code and dataset should I make? Thanks in advance.

ChunyuanLI commented 3 years ago

For pre-training, you don't have to change anything, the default hyper-parameter setting is set to deal with ImageFolders.