Image Segmentation
Transformers
PyTorch
upernet
Inference Endpoints
test2 / tests /test_model.py
mccaly's picture
Upload 660 files
b13b124
import torch
pthfile = r'/SEG/mmsegmentation/pretrained_model/recipe1MP_R50.pth'
net=torch.load(pthfile)
for k in net.keys():
print(k)