noether.training.cli.main_train

Functions

main(config)

Main entry point for training.

Module Contents

noether.training.cli.main_train.main(config)

Main entry point for training.

This script is wrapped in a hydra function to allow for easy configuration. It supports passing the configuration file as a positional argument or via the –hp flag.

Example: .. code-block:: bash

python main_train.py configs/my_experiment.yaml

python main_train.py –hp configs/my_experiment.yaml python main_train.py configs/my_experiment.yaml trainer.max_epochs=10

Parameters:

config (omegaconf.DictConfig)