如何将以下keras的神经网络模型改为pytorch的?

发布于 2021-04-13 09:42:49

这是我想转换为pytorch的示例keras代码!下图为模型结构ex.png
代码如下:

model = Sequential()
model.add(LSTM(input_dim=1, units=50, return_sequences=True))
#model.add(Dropout(0.2))
model.add(LSTM(input_dim=50, units=100, return_sequences=True))
#model.add(Dropout(0.2))
model.add(LSTM(input_dim=100, units=200, return_sequences=True))
#model.add(Dropout(0.2))
model.add(LSTM(300, return_sequences=False))
model.add(Dropout(0.2))
model.add(Dense(100))
model.add(Dense(units=1))
model.add(Activation('relu'))
model.compile(loss='mean_squared_error', optimizer='Adam')
model.summary()

希望有人能解答一下,非常感谢!ex.png

查看更多

关注者
0
被浏览
202
2 个回答
fate
fate 认证专家 2021-04-14

from torch import nn 然后 model = nn.Sequential(),然后后面的一致就行了~

inkknown
inkknown 2021-04-14

试试 onnx 。。感觉模型也没有很重,要不重新训一个呗。。

撰写答案

请登录后再发布答案,点击登录

发布
问题

分享
好友

手机
浏览

扫码手机浏览