Skip to content

Latest commit

 

History

History
10 lines (10 loc) · 467 Bytes

README.md

File metadata and controls

10 lines (10 loc) · 467 Bytes
Raw

Earthformer Training on N-body MNIST

Run the following command to train Earthformer on N-body MNIST dataset. Change the configurations in cfg.yaml

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

Run the tensorboard command to upload experiment records

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