26a33b7
1
2
3
4
import torch a = torch.load('models/pretrained_vit.pth', map_location='cpu') print(a)