/ConvRNNs

Implementation of ConvRNN, ConvLSTM, ConvGRU.

Primary LanguagePython

ConvRNNs

Implementation of ConvLSTM, ConvGRU. Similar to canonical LSTMs and GRUs. However, their fully connected parts have been replaced with convolution operations. ConvLSTM and ConvGRU inherit from BaseConvRNN which is a nn.Module. Therefore, the two classes provided can be used in tandem with other pytorch modules for tasks such as video frame prediction.