Skip to content

Commit

Permalink
Merge pull request #399 from ACEsuit/develop
Browse files Browse the repository at this point in the history
fixed readme typos, and commit
  • Loading branch information
ilyes319 authored Apr 30, 2024
2 parents 8b72edc + a448d36 commit 90f4939
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@
MACE provides fast and accurate machine learning interatomic potentials with higher order equivariant message passing.

This repository contains the MACE reference implementation developed by
Ilyes Batatia, Gregor Simm, and David Kovacs.
Ilyes Batatia, Gregor Simm, David Kovacs, and the group of Gabor Csanyi, and friends (see Contributors).

Also available:

Expand Down Expand Up @@ -215,7 +215,7 @@ python ./mace/scripts/preprocess_data.py \
--seed=123 \
```

To see all options and a little description of them run `python ./mace/scripts/preprocess_data.py --help` . The script will create a number of HDF5 files in the `processed_data` folder which can be used for training. There wiull be one file for trainin, one for validation and a separate one for each `config_type` in the test set. To train the model use the `run_train.py` script as follows:
To see all options and a little description of them run `python ./mace/scripts/preprocess_data.py --help` . The script will create a number of HDF5 files in the `processed_data` folder which can be used for training. There will be one folder for training, one for validation and a separate one for each `config_type` in the test set. To train the model use the `run_train.py` script as follows:

```sh
python ./mace/scripts/run_train.py \
Expand Down
2 changes: 1 addition & 1 deletion mace/cli/run_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -790,7 +790,7 @@ def main() -> None:
model = model.to("cpu")
torch.save(model, model_path)
extra_files = {
"commit.txt": commit.encode("utf-8"),
"commit.txt": commit.encode("utf-8") if commit is not None else b"",
"config.yaml": json.dumps(
convert_to_json_format(extract_config_mace_model(model))
),
Expand Down

0 comments on commit 90f4939

Please sign in to comment.