回调

SpladeRegularizerWeightSchedulerCallback

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

基于计划更新 SpladeLoss 的 query_regularizer_weight 和 document_regularizer_weight 参数的回调。

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

参数：

loss (SpladeLoss) – 要更新的 SpladeLoss 实例
scheduler_type (str) – 调度器类型（“linear” 或 “quadratic”）
warmup_ratio (float) – 达到最大权重值的总步数比率（默认值：1/3）