How to run with dataset

bwconrad / flexivit

PyTorch reimplementation of FlexiViT: One Model for All Patch Sizes

MIT License

48 stars 4 forks source link

How to run with dataset #2

Closed sdeenm closed 1 year ago

sdeenm commented 1 year ago

Hi, sorry for a silly question as I am a new learner in AI, I am unable to run this code with dataset. Could you please give me direction how can I apply flexivit in image classification?

bwconrad commented 1 year ago

This repo only provides an implementation of the FlexiViT architecture and does not contain any training code. You can use the architecture with pretty much any training code repo (for example here) by changing the initialized model to be a FlexiViT.

I did include some evaluation code, however that is only for getting results on the ImageNet validation set to reproduce some experiments from the paper.