PyTorch convert function for op 'fft_rfftn' not implemented

user_18106918 0
Viewed 155

I am trying to convert a pretrained pytorch model to coreml. I am using coremltools to do this. This is my code.

import torch
import coremltools as ct
traced_model = torch.jit.load("a.jit")
model = ct.convert(traced_model, 
          inputs=[ct.TensorType(shape=image.shape), ct.TensorType(shape=mask.shape)])

I am getting this error "PyTorch convert function for op 'fft_rfftn' not implemented." How can I solve this?

coremltools pytorch coreml python ios
2 Answers

Try use my project: RosaKit, there is implemented STFT, it's might help.

user_35519450
user_62444640
Related