diff --git a/modulus/sym/hydra/utils.py b/modulus/sym/hydra/utils.py index 927c7ed8..6937815d 100644 --- a/modulus/sym/hydra/utils.py +++ b/modulus/sym/hydra/utils.py @@ -170,6 +170,7 @@ def compose( register_scheduler_configs() register_training_configs() register_modulus_configs() + register_amp_configs() register_graph_configs() cfg = hydra.compose(