Skip to content

ckjellson/MTO_SW_LSTM

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 

Repository files navigation

Many-to-one sliding window LSTM in Pytorch

  • Many-to-one LSTM using sliding window for arbitrary and varying sequence lengths.
  • GPU-enabled.
  • Uses zero-padding to get an equal number of windows fitted to the sequence lengths using the chosen stride.

Files

  • MTO_SW_LSTM.py - Model class.
  • test_MTO_SW_LSTM.ipynb - Simple demonstration of how to train the MTO_SW_LSTM.

Releases

No releases published

Packages

No packages published