如何将以下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()希望有人能解答一下，非常感谢！
关注者
0
被浏览
1.1k
感谢您的回复，但是这样做的话会提示torch.nn.modules.module.ModuleAttributeError:'Sequential' object has no attribute 'add'。PYtorch里没有像keras里一样的add方法吧。请问这样的话该如何解决呢？