(self)
| 119 | self.load_state_dict(weights) |
| 120 | |
| 121 | def get_gradients(self): |
| 122 | grads = [] |
| 123 | for p in self.parameters(): |
| 124 | grad = None if p.grad is None else p.grad.data.cpu().numpy() |
| 125 | grads.append(grad) |
| 126 | return grads |
| 127 | |
| 128 | def set_gradients(self, gradients): |
| 129 | for g, p in zip(gradients, self.parameters()): |