(self)
| 320 | self.grads[name + '_grad'] += p.grad.data |
| 321 | |
| 322 | def average_gradient(self): |
| 323 | counter_num = self.counter.get() |
| 324 | for name, grad in self.grads.items(): |
| 325 | self.grads[name] /= counter_num |
| 326 | |
| 327 | def print_gradient(self): |
| 328 | for grad in self.grads: |