V2EX = way to explore
V2EX 是一个关于分享和探索的地方
现在注册
已注册用户请  登录
爱意满满的作品展示区。
huntzhan
V2EX  ›  分享创造

Pytorch C++ Extensions 实践: pytorch-stateful-lstm

  •  
  •   huntzhan ·
    huntzhan · 2019-01-09 00:41:09 +08:00 · 1931 次点击
    这是一个创建于 2176 天前的主题,其中的信息可能已经有所发展或是发生改变。
    https://github.com/cnt-dev/pytorch-stateful-lstm

    提供一个 Libtorch & C++ Extension 的 Demo。

    总结:

    - 用相同逻辑把 AllenNLP 的 LSTM Cell 重写了一下,性能提升 50%。
    - 目前 Libtorch 的文档基本没有,而且官方文档与 1.0 的实现已经对不上了,建议想入坑的可以先等等。不过 Libtorch 的源码可读性挺好,没有文档问题不是太大的问题。
    目前尚无回复
    关于   ·   帮助文档   ·   博客   ·   API   ·   FAQ   ·   实用小工具   ·   1440 人在线   最高记录 6679   ·     Select Language
    创意工作者们的社区
    World is powered by solitude
    VERSION: 3.9.8.5 · 21ms · UTC 23:59 · PVG 07:59 · LAX 15:59 · JFK 18:59
    Developed with CodeLauncher
    ♥ Do have faith in what you're doing.