gpt4 book ai didi

python - 如何将 Pytorch 模型导入 MATLAB

转载 作者:行者123 更新时间:2023-12-02 07:24:37 26 4
gpt4 key购买 nike

我已经在 Pytorch 中创建了一个模型,我希望将其转移到 MATLAB,显示了一个最小的示例

import torch.nn as nn
import torch
class cnn(nn.Module):
def __init__(self):
super(cnn, self).__init__()
self.fc1 = nn.Sequential(
nn.Linear(10, 1),
nn.ReLU(True)
)

def forward(self, x):
out = self.fc1(x)
return out
the_net = cnn()
torch.save(the_net,'desperation.h5')

然后在 MATLAB 中调用

net = importKerasLayers('desperation.h5')

这给出了错误消息

Error using importKerasLayers (line 104)
Unable to read HDF5 file 'desperation.h5'. The error message was: 'The filename specified was either
not found on the MATLAB path or it contains unsupported characters.''

该文件位于路径上,我可以将模型加载回 Python。我真正想要的是任何能够让我将模型从 Pytorch 转移到 MATLAB 中而无需手动复制所有权重的解决方案。

我正在运行 MATLAB 2018b、Python 3.6 和 Pytorch 0.4.0

最佳答案

我过去一直在使用这个工具并取得了一些成功:https://github.com/albanie/mcnPyTorch从 Pytorch 转到 MatConvNet。

关于python - 如何将 Pytorch 模型导入 MATLAB,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/51043819/

26 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com