supaernova.steps.pae.model

source module supaernova.steps.pae.model

Classes

• PAEModelStep

source class PAEModelStep[Backend: str](config: PAEModelConfig)

Bases : AbstractModel[Backend]

Methods

• setup_data_masks

source method PAEModelStep.setup_data_masks() → None