search

jeonsworld / ViT-pytorch

Pytorch reimplementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)
MIT License
1.95k stars 374 forks source link

apex version? #49

Closed INF800 closed 2 years ago

INF800 commented 2 years ago

Getting this error with apex==0.9.10.dev0. What is your apex version?

image

INF800 commented 2 years ago

apex==0.1 did the job