(self, mode=True)
| 54 | return self.module(*inputs, **kwargs) |
| 55 | |
| 56 | def train(self, mode=True): |
| 57 | super(DistModule, self).train(mode) |
| 58 | self.module.train(mode) |
| 59 | |
| 60 | def reduce_gradients(self, task_specific=False): |
| 61 | if self.sync: |
no outgoing calls
no test coverage detected