(self, network)
| 172 | layer.parameters.step(n, -step) |
| 173 | |
| 174 | def setup(self, network): |
| 175 | # Accumulators |
| 176 | self.accu = defaultdict(dict) |
| 177 | for i, layer in enumerate(network.parametric_layers): |
| 178 | for n in layer.parameters.keys(): |
| 179 | self.accu[i][n] = np.zeros_like(layer.parameters[n]) |
| 180 | |
| 181 | |
| 182 | class Adam(Optimizer): |