Name		Name	Last commit message	Last commit date
parent directory ..
README.md		README.md
cfg.yaml		cfg.yaml
earthformer_earthnet_v1.yaml		earthformer_earthnet_v1.yaml
inference_tutorial_earthformer_earthnet2021.ipynb		inference_tutorial_earthformer_earthnet2021.ipynb
inference_tutorial_earthformer_earthnet2021x.ipynb		inference_tutorial_earthformer_earthnet2021x.ipynb
predict_cuboid_en21x.py		predict_cuboid_en21x.py
train_cuboid_earthnet.py		train_cuboid_earthnet.py

README.md

Earthformer Training on EarthNet2021 with auxiliary meso scale data

Run the following command to train Earthformer on EarthNet2021 dataset. Change the configurations in cfg.yaml

MASTER_ADDR=localhost MASTER_PORT=10001 python train_cuboid_earthnet.py --gpus 2 --cfg cfg.yaml --ckpt_name last.ckpt --save tmp_earthnet_w_meso

Or run the following command to directly load pretrained checkpoint for test.

MASTER_ADDR=localhost MASTER_PORT=10001 python train_cuboid_earthnet.py --gpus 2 --pretrained --save tmp_earthnet_w_meso

Run the tensorboard command to upload experiment records

tensorboard dev upload --logdir ./experiments/tmp_earthnet_w_meso/lightning_logs --name 'tmp_earthnet_w_meso'