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