Jit.script Tacotron2

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?

is it possible that it’s running out of memory, or something?

1 Like

that’s true, thank you for your suggestion!