Changelog¶
v0.2.0 (21/04/2023)¶
Release version to OpenSTL V0.2.0 as #20.
Code Refactoring¶
Rename the project to
OpenSTLinstead ofSimVPv2with module name refactoring.Refactor the code structure thoroughly to support non-distributed and distributed (DDP) training & testing with
tools/train.pyandtools/test.py.Refactor
_dist_forward_collectand_non_dist_forward_collectto support collection of metrics.
New Features¶
Update the Weather Bench dataloader with
5.625deg,2.8125deg, and1.40625degsettings. Add Human3.6M dataloader (supporting augmentations) and config files. Add Moving FMNIST and MMNIST_CIFAR as two advanced variants of MMNIST datasets.Update tools for dataset preparation of Human3.6M, Weather Bench, and Moving FMNIST.
Support PredNet, TAU, and DMVFN with configs and benchmark results. And fix bugs in these new STL methods.
Support multi-variant versions of Weather Bench with dataloader and metrics.
Support lpips metric for video prediction benchmarks.
Support STL results visualization by vis_video for video prediction, traffic prediction, weather prediction tasks.
Support visualization of STL methods on various datasets (on updating).
Update Documents¶
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/endocuments for the basic usages and new features of V0.2.0. Adding detailed steps of installation and preparation datasets.Clean-up STL benchmarks and update to the latest results with config files provided.
Fix Bugs¶
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_parametersfor DDP training.Fig bugs of building distributed dataloaders and preparation of DDP training.
Fix bugs of some STL methods (CrevNet, DMVFN, PreDNet, and TAU).
Fix bugs in datasets: fixing Caltech dataset for evaluation (28/05/2023 updating Baidu Cloud).
Fix the bug of
PSNR(changing the implementation from E3D-LSTM to the corrent version) and update results in the benchmarks.
v0.1.0 (18/02/2023)¶
Release version to V0.1.0 with code refactoring.
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.
New Features¶
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.
Update Documents¶
Upload
readthedocsdocuments. 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.