Traceback (most recent call last):


 "test.py", line 19, in <module>

model = Pix2PixModel(opt)


 "G:\master\SPADE-master\models\pix2pix_model.py", line 25, in __init__

self.netG, self.netD, self.netE = self.initialize_networks(opt)


 "G:\master\SPADE-master\models\pix2pix_model.py", line 96, in initialize_networks

netG = util.load_network(netG, 'G', opt.which_epoch, opt)


 "G:\master\SPADE-master\util\util.py", line 208, in load_network

net.load_state_dict(weights, strict=False)


 "C:\Users\JMSS\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1045, in load_state_dict

self.__class__.__name__, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for SPADEGenerator:

Traceback (most recent call last):


 "train.py", line 482, in <module>

train(hyp, opt, device, tb_writer, wandb)


 "train.py", line 318, in train

results, maps, times = test.test(opt.data,


 "/workspace/yolov5/test.py", line 125, in test

output = non_max_suppression(inf_out, conf_thres=conf_thres, iou_thres=iou_thres)


 "/workspace/yolov5/utils/general.py", line 661, in non_max_suppression

i = torch.ops.torchvision.nms(boxes, scores, iou_thres)


 "/root/anaconda3/envs/yolo5/lib/python3.8/site-packages/torch/_ops.py", line 61, in __getattr__

op = torch._C._jit_get_operation(qualified_op_name)
RuntimeError: No such operator torchvision::nms
