HHZ94

> Hi > Could you please provide more detail so I can reproduce this error? Such as pytorch version, cuda/cudnn version, test script and anything related? > > By the...

Simple `nn.Module` for test: ``` class Upsample(nn.Module): def __init__(self, scale_factor): super(Upsample, self).__init__() self.scale_factor = scale_factor def forward(self, x): x_size = (x.size(-2), x.size(-1)) original_x_step = 1 / (x_size[-2] - 1) x_grid...

Driver Version: 440.64 CUDA Version: 10.2 CUDNN: 8.0.5 torch 1.5.0 TensorRT 7.2.2.3

Remove this line ```angles = torch.abs(angles)*torch.sign(cam_new[:,:3])``` And everything goes well.