(self, opt_idx)
| 161 | } |
| 162 | |
| 163 | def on_before_optimization(self, opt_idx): |
| 164 | if self.gradient_clip_norm > 0: |
| 165 | torch.nn.utils.clip_grad_norm_(self.parameters(), self.gradient_clip_norm) |
| 166 | if self.gradient_clip_val > 0: |
| 167 | torch.nn.utils.clip_grad_value_(self.parameters(), self.gradient_clip_val) |
| 168 | |
| 169 | def on_after_optimization(self, epoch, batch_idx, optimizer, optimizer_idx): |
| 170 | if self.scheduler is not None: |