diff --git a/README.md b/README.md index 2cee74e..aaec835 100644 --- a/README.md +++ b/README.md @@ -174,7 +174,7 @@ This will take some time (~1h). You will get preprocessed data files in `dataset You can train the MatterGen base model on `mp_20` using the following command. ```bash -python scripts/run.py data_module=mp_20 ~trainer.logger ~trainer.strategy trainer.accelerator=mps +python scripts/run.py data_module=mp_20 ~trainer.logger ``` The validation loss (`loss_val`) should reach 0.4 after 360 epochs (about 80k steps). The output checkpoints can be found at `outputs/singlerun/${now:%Y-%m-%d}/${now:%H-%M-%S}`. We call this folder `$MODEL_PATH` for future reference.