test / FoodSeg103 /tests /test_model.py
mccaly's picture
Upload 2805 files
985cc7f
raw
history blame contribute delete
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)