(self, sd)
| 75 | return sd |
| 76 | |
| 77 | def load_state_dict(self, sd): |
| 78 | # self.start_lr = sd['start_lr'] |
| 79 | self.warmup_iter = sd['warmup_iter'] |
| 80 | self.num_iters = sd['num_iters'] |
| 81 | # self.end_iter = sd['end_iter'] |
| 82 | # self.decay_style = sd['decay_style'] |
| 83 | # if 'decay_ratio' in sd: |
| 84 | # self.decay_ratio = sd['decay_ratio'] |
| 85 | self.step(self.num_iters) |
| 86 | |
| 87 | def switch_linear(self, args): |
| 88 | current_lr = self.get_lr() |