File size: 86 Bytes
26a33b7
 
 
 
1
2
3
4
import torch

a = torch.load('models/pretrained_vit.pth', map_location='cpu')
print(a)