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

2019-01-09 00:41:09 +08:00
 huntzhan
https://github.com/cnt-dev/pytorch-stateful-lstm

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

总结:

- 用相同逻辑把 AllenNLP 的 LSTM Cell 重写了一下,性能提升 50%。
- 目前 Libtorch 的文档基本没有,而且官方文档与 1.0 的实现已经对不上了,建议想入坑的可以先等等。不过 Libtorch 的源码可读性挺好,没有文档问题不是太大的问题。
1935 次点击
所在节点    分享创造
0 条回复

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/525208

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX