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

aijiyenan 1月前 23

这是我想转换为 pytorch 的示例 keras 代码!下图为模型结构 https://file.nuomiphp.com/uploads/20210413/FmdyObK_bDET2ChmVw44z33QaYGO.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()

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

最新回复 (6)
  • GrayXu 26天前
    引用 2
    试试 onnx 。。感觉模型也没有很重,要不重新训一个呗。。
  • mikumkf 26天前
    引用 3
    from torch import nn 然后 model = nn.Sequential(),然后后面的一致就行了~
  • jc89898 26天前
    引用 4
    这么简单的网络,你手写一个也要不了 10 分钟吧,而且基本都是一样的代码。
  • 楼主 aijiyenan 26天前
    引用 5
    @GrayXu
    @mikumkf
    @jc89898 谢过大佬们,现在用 pytorch 重新写了一个。
  • Yeen 26天前
    引用 6
    谁有免费的 gpu 服务器推荐的,只需要试用一段时间也可以。
  • xyfan 26天前
    引用 7
    @Yeen 看你想用多久了,简单测试用 Colab
  • 游客
    8
返回