PyTorch-Pretrained-ViT
PyTorch-Pretrained-ViT copied to clipboard
Update model.py
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.