PyTorch模型转CoreML提示type_as算子未实现错误如何解决

RuntimeError: PyTorch convert function for op 'type_as' not implemented. 说明你当前使用的coremltools版本暂不支持对PyTorch的type_as算子进行转换，该算子的作用是将一个张量的数据类型转换为和另一个张量相同的类型，你的模型内部逻辑中调用了这个算子。

• 方案1：升级coremltools到最新版本
  coremltools的新版本会陆续补充对未支持算子的适配，你可以先尝试执行更新命令：
  pip install --upgrade coremltools
  更新完成后重新运行转换代码排查问题。
• 方案2：替换模型中的type_as算子
  如果升级后仍然不支持，你可以修改PyTorch模型的源码，将所有type_as调用替换为等价的、coremltools支持的写法，比如将a = a.type_as(b)替换为：
  a = a.to(b.dtype)
  coremltools对to操作的适配更加完善，绝大多数版本都支持该算子的转换。修改完成后重新导出.pt模型，再进行CoreML转换即可。
• 方案3：使用PyTorch脚本模式代替追踪模式
  部分情况下torch.jit.trace追踪生成的计算图会额外引入type_as算子，你可以尝试改用脚本模式导出模型：

# 替换原来的trace行
script = torch.jit.script(model)
# 后续用script代替trace传入ct.convert
mlmodel = ct.convert(script, inputs=[ct.ImageType(name="input_1", shape=input_batch.shape, bias=[1, 0.2/0.3, 0.1/0.7], scale = 1./(255*0.67))])

另外你当前的预处理代码中存在笔误：transforms.Normalize(mean=[0.5, 0,2, 0.1]里的第二个均值参数把.写成了,，会导致预处理报错，需要修正为0.2。

内容的提问来源于stack exchange，提问作者Aditya Rudrapatna