code: def __init__(self, optimizer, t_max, eta_min=0, last_epoch=-1):
      self.last_epoch = last_epoch
      self.base_wds = [
          group["weight_decay"] for group in optimizer.param_groups
      ]
      self.t_max = t_max
      self.eta_min = eta_min
  
imports: import math
label_desc: Use image node_id 1 to create a new CosineWDSchedule object  with example: obj = CosineWDSchedule(optimizer, t_max, eta_min, last_epoch)