trained_model = Net()
m = torch.load(‘./statenew.pth’)
trained_model.load_state_dict(m)
model1 = trained_model.eval().cuda()
x= torch.ones((1,3,100,100)).cuda()
model_trt = torch2trt(model1,)
I am getting an error :
Traceback (most recent call last):
File “nanoconv.py”, line 29, in
model_trt = torch2trt(model1,)
File “/usr/local/lib/python3.6/dist-packages/torch2trt-0.0.0-py3.6.egg/torch2trt/torch2trt.py”, line 241, in torch2trt
outputs = module(*inputs)
File “/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py”, line 493, in call
result = self.forward(*input, **kwargs)
File “/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py”, line 88, in forward
raise NotImplementedError
NotImplementedError