python - 如何将 pytorch 模型转换为核心 ml 模型？

Question

我训练了一个模型并得到了一个名为 model_final.pth 的输出。

我加载模型如下。

#load model
model = torch.load('/content/drive/MyDrive/model/model_final.pth')

我按照以下步骤操作。

#load sample image
input_image = Image.open("/content/drive/MyDrive/model/g0079.jpg")
display(input_image)

to_tensor = transforms.ToTensor()
input_tensor = to_tensor(input_image)
input_batch = input_tensor.unsqueeze(0)

然后当我运行下面的代码时，我得到一个错误。

#First attempt at tracing
trace = torch.jit.trace(model, input_batch)

错误：

为什么我会收到此错误？你觉得我应该怎么做？

标签： pythonmachine-learningdeep-learningpytorchcoreml