Method step

utils/general_utils.py:233–240

(self, epoch=None)

        return [wd for _ in self.optimizer.param_groups]

    def step(self, epoch=None):
        # 先让父类更新 last_epoch
        self._step_count += 1
        # super().step(epoch)
        # 再把新的 weight_decay 写到 optimizer
        new_wd = self.get_lr()
        for group, wd in zip(self.optimizer.param_groups, new_wd):
            group['weight_decay'] = wd