🎯 Production-ready implementation of video prediction models using PyTorch. Features Enhanced ConvLSTM with temporal attention, PredRNN with spatiotemporal memory, and Transformer-based architecture.
deep-learning pytorch transformer lstm attention-mechanism ucf101 convlstm video-prediction gradient-accumulation temporal-attention predrnn spatiotemporal-memory
-
Updated
Dec 23, 2024 - Python