(self, grad_clip=None, coalesce=True, bucket_size_mb=-1)
| 47 | class DistOptimizerHook(OptimizerHook): |
| 48 | |
| 49 | def __init__(self, grad_clip=None, coalesce=True, bucket_size_mb=-1): |
| 50 | self.grad_clip = grad_clip |
| 51 | self.coalesce = coalesce |
| 52 | self.bucket_size_mb = bucket_size_mb |
| 53 | |
| 54 | def after_train_iter(self, runner): |
| 55 | runner.optimizer.zero_grad() |
nothing calls this directly
no outgoing calls
no test coverage detected