test2 / tests /test_model.py
mccaly's picture
Upload 660 files
b13b124
raw history blame
No virus
142 Bytes
import torch
pthfile = r'/SEG/mmsegmentation/pretrained_model/recipe1MP_R50.pth'
net=torch.load(pthfile)
for k in net.keys():
print(k)