← all repositories

ndrplz/ConvLSTM_pytorch

A PyTorch module implementing Convolutional LSTM for spatiotemporal sequence modeling.

2.3k stars Python ML Frameworks
ConvLSTM_pytorch
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of the ConvLSTM cell, a deep learning architecture that combines convolutional operations with LSTM gates for processing spatiotemporal data. The module supports stacking arbitrary numbers of layers with configurable hidden dimensions (channel counts) and kernel sizes. It derives from nn.Module and can be integrated into larger neural network architectures for tasks like video prediction, weather forecasting, and traffic flow modeling.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.