Initialize optimizer state
(self, parameter: mx.array, state: dict)
| 266 | self.nesterov = nesterov |
| 267 | |
| 268 | def init_single(self, parameter: mx.array, state: dict): |
| 269 | """Initialize optimizer state""" |
| 270 | state["v"] = mx.zeros_like(parameter) |
| 271 | |
| 272 | def apply_single(self, gradient: mx.array, parameter: mx.array, state: dict): |
| 273 | """Performs the SGD parameter update and stores :math:`v` in the |
nothing calls this directly
no outgoing calls
no test coverage detected