None
Project description
pytorch-stateful-lstm
Free software: MIT license
Features
Pytorch LSTM implementation powered by Libtorch, and with the support of:
Hidden/Cell Clip.
Variational Dropout & DropConnect.
Managed Initial State.
Built-in TBPTT.
Benchmark: https://github.com/cnt-dev/pytorch-stateful-lstm/tree/master/benchmark
Install
Prerequisite: torch>=1.0.0, supported C++11 compiler (see here). To install through pip:
pip install pytorch-stateful-lstm
Usage
Example:
from pytorch_stateful_lstm import StatefulUnidirectionalLstm lstm = StatefulUnidirectionalLstm( num_layers=2, input_size=3, hidden_size=5, cell_size=7, ) output_sequence, lstm_state = lstm( input_tensor, [5, 4, 2, 1], )
For the definition of parameters, see https://github.com/cnt-dev/pytorch-stateful-lstm/tree/master/extension.
Credits
This package was created with Cookiecutter and the audreyr/cookiecutter-pypackage project template.
History
0.1.0 (2019-01-03)
First release on PyPI.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Close
Hashes for pytorch_stateful_lstm-1.1.1.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | eb21d5b862bae1baf006a1cfcf0714cdd4060459b51b03b1957dbdf3696cbb4c |
|
MD5 | aeafbaa0dcd8836ceb996edc3bf6c3a6 |
|
BLAKE2b-256 | 7670a649e08f4909e77c668dce0b1e0d89ee3ff7a74405ff55089aaf61d4352a |