deepmd_gnn.argcheck

deepmd_gnn.argcheck#

Argument check for the MACE model.

Functions#

mace_model_args(→ dargs.Argument)

Arguments for the MACE model.

nequip_model_args(→ dargs.Argument)

Arguments for the NequIP model.

Module Contents#

deepmd_gnn.argcheck.mace_model_args() → dargs.Argument[source]#

Arguments for the MACE model.

Returns#

Argument

Arguments for the MACE model.

deepmd_gnn.argcheck.nequip_model_args() → dargs.Argument[source]#

Arguments for the NequIP model.