LICENSE
MANIFEST.in
README.md
requirements.txt
setup.cfg
setup.py
earthformer/__init__.py
earthformer/common.py
earthformer/config.py
earthformer/earthformer.py
earthformer/train_cuboid_earthnet.py
earthformer.egg-info/PKG-INFO
earthformer.egg-info/SOURCES.txt
earthformer.egg-info/dependency_links.txt
earthformer.egg-info/not-zip-safe
earthformer.egg-info/requires.txt
earthformer.egg-info/top_level.txt
earthformer/baselines/__init__.py
earthformer/baselines/persistence.py
earthformer/cuboid_transformer/__init__.py
earthformer/cuboid_transformer/cuboid_transformer.py
earthformer/cuboid_transformer/cuboid_transformer_patterns.py
earthformer/cuboid_transformer/cuboid_transformer_unet_dec.py
earthformer/cuboid_transformer/utils.py
earthformer/datasets/__init__.py
earthformer/datasets/augmentation.py
earthformer/datasets/earthnet/__init__.py
earthformer/datasets/earthnet/earthnet21x_dataloader.py
earthformer/datasets/earthnet/earthnet_dataloader.py
earthformer/datasets/earthnet/earthnet_scores.py
earthformer/datasets/earthnet/visualization.py
earthformer/datasets/earthnet/earthnet_toolkit/__init__.py
earthformer/datasets/earthnet/earthnet_toolkit/coords.py
earthformer/datasets/earthnet/earthnet_toolkit/coords_dict.py
earthformer/datasets/earthnet/earthnet_toolkit/download.py
earthformer/datasets/earthnet/earthnet_toolkit/download_links.py
earthformer/datasets/earthnet/earthnet_toolkit/parallel_score.py
earthformer/datasets/earthnet/earthnet_toolkit/plot_cube.py
earthformer/datasets/enso/__init__.py
earthformer/datasets/enso/enso_dataloader.py
earthformer/datasets/moving_mnist/__init__.py
earthformer/datasets/moving_mnist/moving_mnist.py
earthformer/datasets/nbody/__init__.py
earthformer/datasets/nbody/nbody_mnist.py
earthformer/datasets/nbody/nbody_mnist_torch_wrap.py
earthformer/datasets/sevir/__init__.py
earthformer/datasets/sevir/sevir_dataloader.py
earthformer/datasets/sevir/sevir_torch_wrap.py
earthformer/metrics/__init__.py
earthformer/metrics/enso.py
earthformer/metrics/sevir.py
earthformer/metrics/skill_scores.py
earthformer/metrics/torchmetrics_wo_compute.py
earthformer/scripts/baselines/persistence/earthnet/README.md
earthformer/scripts/baselines/persistence/earthnet/cfg.yaml
earthformer/scripts/baselines/persistence/earthnet/test_persistence_earthnet.py
earthformer/scripts/cuboid_transformer/earthnet_w_meso/README.md
earthformer/scripts/cuboid_transformer/earthnet_w_meso/cfg.yaml
earthformer/scripts/cuboid_transformer/earthnet_w_meso/earthformer_earthnet_v1.yaml
earthformer/scripts/cuboid_transformer/earthnet_w_meso/predict_cuboid_en21x.py
earthformer/scripts/cuboid_transformer/earthnet_w_meso/train_cuboid_earthnet.py
earthformer/scripts/cuboid_transformer/enso/README.md
earthformer/scripts/cuboid_transformer/enso/cfg.yaml
earthformer/scripts/cuboid_transformer/enso/earthformer_enso_v1.yaml
earthformer/scripts/cuboid_transformer/enso/train_cuboid_enso.py
earthformer/scripts/cuboid_transformer/moving_mnist/README.md
earthformer/scripts/cuboid_transformer/moving_mnist/cfg.yaml
earthformer/scripts/cuboid_transformer/moving_mnist/train_cuboid_mnist.py
earthformer/scripts/cuboid_transformer/nbody/README.md
earthformer/scripts/cuboid_transformer/nbody/cfg.yaml
earthformer/scripts/cuboid_transformer/nbody/train_cuboid_nbody.py
earthformer/scripts/cuboid_transformer/sevir/README.md
earthformer/scripts/cuboid_transformer/sevir/cfg_sevir.yaml
earthformer/scripts/cuboid_transformer/sevir/cfg_sevirlr.yaml
earthformer/scripts/cuboid_transformer/sevir/earthformer_sevir_v1.yaml
earthformer/scripts/cuboid_transformer/sevir/train_cuboid_sevir.py
earthformer/scripts/datasets/nbody/README.md
earthformer/scripts/datasets/nbody/cfg.yaml
earthformer/scripts/datasets/nbody/download_nbody_paper.py
earthformer/scripts/datasets/nbody/generate_nbody_dataset.py
earthformer/scripts/datasets/sevir/download_sevir.py
earthformer/utils/__init__.py
earthformer/utils/apex_ddp.py
earthformer/utils/checkpoint.py
earthformer/utils/layout.py
earthformer/utils/optim.py
earthformer/utils/registry.py
earthformer/utils/utils.py
earthformer/visualization/__init__.py
earthformer/visualization/nbody.py
earthformer/visualization/utils.py
earthformer/visualization/sevir/__init__.py
earthformer/visualization/sevir/dataset_statistics.py
earthformer/visualization/sevir/sevir_cmap.py
earthformer/visualization/sevir/sevir_vis_seq.py