#candle #lstm #pytorch

candle-lstm

在某些情况下优化 HuggingFace candle LSTM

2 个不稳定版本

新版本 0.2.0 2024 年 8 月 25 日
0.1.0 2024 年 8 月 24 日

#544机器学习

Download history 250/week @ 2024-08-19

每月下载量 250

自定义许可证

23KB
511

Candle LSTM

重新实现 Candle LSTM 推理以提高推理速度,包括双向 LSTM。

测试数据

安装 Pytorch 并运行 simple.py 生成测试数据。

  1. lstm_test.pt: Pytorch LSTM,batch_first = False。
  2. lstm_test_batch_first.pt: Pytorch LSTM,batch_first = True。
  3. bi_lstm_test.pt: Pytorch 双向 LSTM,batch_first = False。
  4. bi_lstm_test_batch_first.pt: Pytorch 双向 LSTM,batch_first = True。

依赖关系

~11MB
~218K SLoC