Hello, i have successfully managed to convert NVIDIA Tacotron2 using torch.jit.script. However, when i try to run it on an Android Device i get a RuntimeError: NNPACK SpatialConvolution_updateOutput failed
Below is the exact error:
2020-05-29 16:25:58.166 24263-24427…/AndroidRuntime:   File “…/python3.6/site-packages/torch/nn/modules/conv.py”, line 207, in forward
self.weight, self.bias, self.stride,
_single(0), self.dilation, self.groups)
return F.conv1d(input, self.weight, self.bias, self.stride,
~~~~~~~~ <— HERE
self.padding, self.dilation, self.groups)
RuntimeError: NNPACK SpatialConvolution_updateOutput failed
Jit is created with PyTorch 1.5 and python 3.6.10
The Android version is 9.
Any idea on how to resolve this problem?