Update model.py - Githubissues

lukemelas / PyTorch-Pretrained-ViT

Vision Transformer (ViT) in PyTorch

770 stars 124 forks source link

Update model.py #13

Open nathansomavarapu opened 3 years ago

nathansomavarapu commented 3 years ago

I think that this line is what is stopping a user from creating a custom model. I don't think the referenced variable is available in the scope that it is being used.