(self, batch, cond)
| 170 | self.save() |
| 171 | |
| 172 | def run_step(self, batch, cond): |
| 173 | self.forward_backward(batch, cond) |
| 174 | took_step = self.mp_trainer.optimize(self.opt) |
| 175 | if took_step: |
| 176 | self._update_ema() |
| 177 | self._anneal_lr() |
| 178 | self.log_step() |
| 179 | |
| 180 | def forward_backward(self, batch, cond): |
| 181 | self.mp_trainer.zero_grad() |
no test coverage detected