nemo_automodel.components.training

Training utilities shared across recipes.

Submodules

• nemo_automodel.components.training.ema
• nemo_automodel.components.training.garbage_collection
• nemo_automodel.components.training.model_output_utils
• nemo_automodel.components.training.neftune
• nemo_automodel.components.training.rng
• nemo_automodel.components.training.signal_handler
• nemo_automodel.components.training.step_scheduler
• nemo_automodel.components.training.timers
• nemo_automodel.components.training.utils

Package Contents

Data

__all__

API

nemo_automodel.components.training.__all__ = ['StepSchedulerConfig']