Release version to OpenSTL V0.2.0 as #20.
- Rename the project to
OpenSTL
instead ofSimVPv2
with module name refactoring. - Refactor the code structure thoroughly to support non-distributed and distributed (DDP) training & testing with
tools/train.py
andtools/test.py
.
- Update the Weather Bench dataloader with
5.625deg
,2.8125deg
, and1.40625deg
settings. Add Human3.6M dataloader (supporting augmentations) and config files. Add Moving FMNIST in as an advanced variants of MMNIST datasets. - Update tools for dataset preparation of Human3.6M, Weather Bench, and Moving FMNIST.
- Update documents of video prediction, traffic prediction, and weather prediction benchmarks with benchmark results and spesific GPU settings (e.g., single GPU). Provide config files for supported STL methods.
- Update
docs/en
documents for the basic usages and new features of V0.2.0.
- Fix bugs in training loops and validation loops to save GPU memory.
- There might be some bugs in not using all parameters for calculating losses in ConvLSTM CrevNet, which should use
--find_unused_parameters
for DDP training. - Fig bugs of building distributed dataloaders and preparation of DDP training.
- Fix bugs of some STL methods (CrevNet and PreDNet).
Release version to V0.1.0 with code refactoring.
- Refactor code structures as
simvp/api
,simvp/core
,simvp/datasets
,simvp/methods
,simvp/models
,simvp/modules
. We support non-distributed training and evaluation by the executable python filetools/non_dist_train.py
. Refactor config files for SimVP models. - Fix bugs in tools/nondist_train.py, simvp/utils, environment.yml, and .gitignore, etc.
- Support Timm optimizers and schedulers.
- Update popular Metaformer models as the hidden Translator
$h$ in SimVP, supporting ViT, Swin-Transformer, MLP-Mixer, ConvMixer, UniFormer, PoolFormer, ConvNeXt, VAN, HorNet, and MogaNet. - Update implementations of dataset and dataloader, supporting KTH Action, KittiCaltech Pedestrian, Moving MNIST, TaxiBJ, and WeatherBench.
- Upload
readthedocs
documents. Summarize video prediction benchmark results on MMNIST in video_benchmarks.md. - Update benchmark results of video prediction baselines and MetaFormer architectures based on SimVP on MMNIST, TaxiBJ, and WeatherBench datasets.
- Update README and add a license.