nemo_automodel.components.training.garbage_collection

Module Contents

Classes

GarbageCollection

Utility for periodic manual garbage collection during training.

Data

logger

API

nemo_automodel.components.training.garbage_collection.logger

‘getLogger(…)’

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

Utility for periodic manual garbage collection during training.

Initialization

run(step_count: int) → None

Run periodic garbage collection based on step count.

static collect(reason: str, generation: int = 1) → None

Collect garbage and emit timing logs.