Please, could anyone help me check where I did wrong?
When I run, an error occurs. I can’t solve it.
please any help or suggestions would be highly appreciated
model = torch.load('./media/model_best.pth')
model = torch.nn.DataParallel(model['model_def'](20))
model.load_state_dict(model['state_dict'])
model.cuda()
model.eval()

image1005×380 53.9 KB