Method __init__

utils/general_utils.py:196–217

对 weight_decay 进行余弦“增加”调度，从 initial_wd -> final_wd。 参数: - optimizer: 任何带有 'weight_decay' param_group 的 optimizer - max_iters: 总的调度步数 - initial_wd: 第 0 步时的 weight_decay - final_wd: 第 max_iters 步时的 weight_decay - last_epoch: 如需从中途恢复训练，传

(
        self,
        optimizer,
        max_iters: int,
        initial_wd: float = 0.05,
        final_wd: float = 0.20,
        last_epoch: int = -1,
    )

class CosineWeightDecayScheduler(LRScheduler):
    def __init__(
        self,
        optimizer,
        max_iters: int,
        initial_wd: float = 0.05,
        final_wd: float = 0.20,
        last_epoch: int = -1,
    ):
        """
        对 weight_decay 进行余弦“增加”调度，从 initial_wd -> final_wd。
        
        参数:
        - optimizer: 任何带有 'weight_decay' param_group 的 optimizer
        - max_iters: 总的调度步数
        - initial_wd: 第 0 步时的 weight_decay
        - final_wd: 第 max_iters 步时的 weight_decay
        - last_epoch: 如需从中途恢复训练，传入上次迭代 idx
        """
        self.max_iters = max_iters
        self.initial_wd = initial_wd
        self.final_wd = final_wd
        super().__init__(optimizer, last_epoch)

    def get_lr(self):
        step = self._step_count