(self, runner)
| 51 | self.bucket_size_mb = bucket_size_mb |
| 52 | |
| 53 | def after_train_iter(self, runner): |
| 54 | runner.optimizer.zero_grad() |
| 55 | runner.outputs['loss'].backward() |
| 56 | if self.grad_clip is not None: |
| 57 | self.clip_grads(runner.model.parameters()) |
| 58 | runner.optimizer.step() |
| 59 | |
| 60 | |
| 61 | def reduce_mean(tensor): |
nothing calls this directly
no test coverage detected