nemo_automodel.components.training.garbage_collection

Module Contents

Name	Description
`GarbageCollection`	Utility for periodic manual garbage collection during training.

class nemo_automodel.components.training.garbage_collection.GarbageCollection(
    gc_every_steps: int = 1000
)

Utility for periodic manual garbage collection during training.

nemo_automodel.components.training.garbage_collection.GarbageCollection.collect(
    reason: str,
    generation: int = 1
) -> None

staticmethod

Collect garbage and emit timing logs.

nemo_automodel.components.training.garbage_collection.GarbageCollection.run(
    step_count: int
) -> None

Run periodic garbage collection based on step count.

nemo_automodel.components.training.garbage_collection.logger = logging.getLogger(__name__)