Thanks for sharing your ConvLSTM implementation! Could you still share the BaseModule class that the ConvLSTMCell inherits? Currently, the description for the forward pass of the LSTM cell is missing. BR, mikko