Pytorch C++ Extensions 实践： pytorch-stateful-lstm
huntzhan · huntzhan · 2019-01-09 00:41:09 +08:00 · 1935 次点击这是一个创建于 2180 天前的主题，其中的信息可能已经有所发展或是发生改变。
https://github.com/cnt-dev/pytorch-stateful-lstm
提供一个 Libtorch & C++ Extension 的 Demo。
总结：
- 用相同逻辑把 AllenNLP 的 LSTM Cell 重写了一下，性能提升 50%。
- 目前 Libtorch 的文档基本没有，而且官方文档与 1.0 的实现已经对不上了，建议想入坑的可以先等等。不过 Libtorch 的源码可读性挺好，没有文档问题不是太大的问题。
提供一个 Libtorch & C++ Extension 的 Demo。
总结：
- 用相同逻辑把 AllenNLP 的 LSTM Cell 重写了一下，性能提升 50%。
- 目前 Libtorch 的文档基本没有，而且官方文档与 1.0 的实现已经对不上了，建议想入坑的可以先等等。不过 Libtorch 的源码可读性挺好，没有文档问题不是太大的问题。
目前尚无回复
Select Language