SR-Forge Loss Scheduler¶ LossScheduler ¶ Epoch-aware loss wrapper that swaps the active Metric at configured milestones. Parameters: Name Type Description Default schedule Mapping[int, Metric] Mapping of {epoch_milestone: Metric}. 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() -> Metric ¶ Return the currently active Metric instance. Was this page helpful? Thanks for your feedback! Thanks! Help us improve by opening an issue.