torchcell.losses.mle_dist_supcr.TemperatureScheduler

class TemperatureScheduler(init_temp: float = 1.0, final_temp: float = 0.1, schedule: str = 'exponential')

Bases: object

Manages temperature scheduling for SupCR loss.

schedule must be one of TEMPERATURE_SCHEDULES (“cosine”, “exponential”); any other name raises ValueError at construction.

get_temperature(epoch: int, max_epochs: int = 1000) → float

Get temperature based on training epoch.