Closed chengWr closed 5 years ago
print('Loading model..') net = RetinaNet() net.load_state_dict(torch.load('./checkpoint/params.pth')['net']) net.eval()
print('Loading model..') net = RetinaNet() net.load_state_dict(torch.load('./checkpoint/params.pth')['net']) net.eval()