Training
mlcg provides some tools to train its models in the scripts folder and some example input files such as examples/train_schnet.yaml. The training is defined using the pytorch-lightning package and especially its cli utilities.
Scripts
Scripts that are using LightningCLI have many convinient builtin functionalities such as a detailed helper
python scripts/mlcg-train.py --help