V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
aijiyenan
V2EX  ›  程序员

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

  •  
  •   aijiyenan · 2021-04-14 10:43:35 +08:00 · 1214 次点击
    这是一个创建于 1079 天前的主题,其中的信息可能已经有所发展或是发生改变。

    这是我想转换为 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 条回复    2021-04-14 16:28:45 +08:00
    GrayXu
        1
    GrayXu  
       2021-04-14 11:02:36 +08:00
    试试 onnx 。。感觉模型也没有很重,要不重新训一个呗。。
    mikumkf
        2
    mikumkf  
       2021-04-14 11:05:59 +08:00
    from torch import nn 然后 model = nn.Sequential(),然后后面的一致就行了~
    jc89898
        3
    jc89898  
       2021-04-14 11:06:18 +08:00
    这么简单的网络,你手写一个也要不了 10 分钟吧,而且基本都是一样的代码。
    aijiyenan
        4
    aijiyenan  
    OP
       2021-04-14 14:15:10 +08:00
    @GrayXu
    @mikumkf
    @jc89898 谢过大佬们,现在用 pytorch 重新写了一个。
    Yeen
        5
    Yeen  
       2021-04-14 15:21:11 +08:00
    谁有免费的 gpu 服务器推荐的,只需要试用一段时间也可以。
    xyfan
        6
    xyfan  
       2021-04-14 16:28:45 +08:00
    @Yeen 看你想用多久了,简单测试用 Colab
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   我们的愿景   ·   实用小工具   ·   1732 人在线   最高记录 6543   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 24ms · UTC 16:39 · PVG 00:39 · LAX 09:39 · JFK 12:39
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.