测试时遇到问题 - Githubissues

kuangliu / pytorch-retinanet

RetinaNet in PyTorch

992 stars 250 forks source link

Closed chengWr closed 5 years ago

chengWr commented 5 years ago

print('Loading model..') net = RetinaNet() net.load_state_dict(torch.load('./checkpoint/params.pth')['net']) net.eval()