这是一个创建于 1318 天前的主题，其中的信息可能已经有所发展或是发生改变。
这是我想转换为 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()
希望有人能解答一下，非常感谢！
 |
1
GrayXu 2021-04-14 11:02:36 +08:00
试试 onnx 。。感觉模型也没有很重，要不重新训一个呗。。
|
 |
2
mikumkf 2021-04-14 11:05:59 +08:00
from torch import nn 然后 model = nn.Sequential()，然后后面的一致就行了~
|
 |
3
jc89898 2021-04-14 11:06:18 +08:00
这么简单的网络，你手写一个也要不了 10 分钟吧，而且基本都是一样的代码。
|
 |
5
Yeen 2021-04-14 15:21:11 +08:00
谁有免费的 gpu 服务器推荐的，只需要试用一段时间也可以。
|
Select Language