code: def update_weight_decay(self, optimizer):
      self.last_epoch += 1
      values = self._get_wd(optimizer)
      for i, data in enumerate(zip(optimizer.param_groups, values)):
          param_group, wd = data
          # Avoid updating weight decay of param_groups that should not be decayed.
          if param_group["weight_decay"] > 0.0:
              param_group["weight_decay"] = wd
  
imports: import math
label_desc: Use image node_id 3 for calling the CosineWDSchedule obj's underlying member method code with example usage: obj.update_weight_decay(optimizer) without return types