osmr / imgclsmob

Sandbox for training deep learning networks

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Visual Transformers?

Bekhouche opened this issue · comments

Are there any plans to add some ViT models with their pretrained weights (ImageNet)?
https://github.com/lucidrains/vit-pytorch