The code in Example Classification is wrong?

lukemelas / PyTorch-Pretrained-ViT

Vision Transformer (ViT) in PyTorch

770 stars 124 forks source link

Closed GitGhidorah closed 3 years ago

GitGhidorah commented 3 years ago

Hello!

i think the code in Example Classification is wrong. Im correct?

Example: The position of "model.eval()" in codes is correct?

GitGhidorah commented 3 years ago

Sorry! Im wrong. The position of "model.eval()" is right.