SR-Forge Loss Scheduler¶ LossScheduler ¶ Epoch-aware loss wrapper that swaps the active Loss at configured milestones. Parameters: Name Type Description Default schedule Mapping[int, Loss] Mapping of {epoch_milestone: Loss}. Must contain key 0. required start_epoch int Current/resume epoch (defaults to 0). 0 __call__(*args, **kwargs) -> MetricScores ¶ Delegate to the currently active loss function. update(epoch: int) -> LossScheduler ¶ Advance the scheduler to epoch, updating the active loss. get_loss_fn() -> Loss ¶ Return the currently active Loss instance. Was this page helpful? Thanks for your feedback! Thanks! Help us improve by opening an issue.