Skip to content

Latest commit

 

History

History

nbody

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
README.md
README.md
 
 
cfg.yaml
cfg.yaml
 
 
train_cuboid_nbody.py
train_cuboid_nbody.py
 
 

README.md

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'