torchcell.losses.mle_wasserstein.TemperatureScheduler

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

Bases: object

Manages temperature scheduling for SupCR loss.

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

Get temperature based on training epoch.