回调

SpladeRegularizerWeightSchedulerCallback

class sentence_transformers.sparse_encoder.callbacks.splade_callbacks.SpladeRegularizerWeightSchedulerCallback(loss: SpladeLoss, scheduler_type: str | SchedulerType = SchedulerType.QUADRATIC, warmup_ratio: float = 0.3333333333333333)[源代码]

此回调根据一个调度计划更新 SpladeLoss 的 query_regularizer_weight 和 document_regularizer_weight 参数。

调度器在总训练步数的指定预热比例内，将权重值从 0 逐渐增加到其最大值。

参数:

  • loss (SpladeLoss) – 需要更新的 SpladeLoss 实例

  • scheduler_type (str) – 调度器类型（'linear' 或 'quadratic'）

  • warmup_ratio (float) – 达到最大权重值所需的总步数比例（默认为 1/3）