ACEsuit / mace

MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training flag not working

ericyuan00000 opened this issue · comments

Describe the bug

When running

python <mace_repo_dir>/mace/cli/run_train.py ... --pin_memory=False

the flag pin_memory remains True.

The code

parser.add_argument(
    "--pin_memory",
    help="Pin memory for data loading",
    default=True,
    type=bool,
)

in arg_parser.py does not seem to be the correct syntax.

If the default is True, you could use "--no_pin_memory", dest="pin_memory", action="store_false"?

Also, some additional alternatives including some more modern (py >= 3.9 specific) in https://stackoverflow.com/questions/15008758/parsing-boolean-values-with-argparse