minor fix in optimizer + docs (#1264)

2025-09-19 10:48:09 +08:00 · 2024-07-12 12:18:02 -07:00
parent 218047c75a
commit 8c01a7893b
2 changed files with 36 additions and 0 deletions
--- a/python/mlx/optimizers/optimizers.py
+++ b/python/mlx/optimizers/optimizers.py
@@ -104,6 +104,7 @@ class Optimizer:

    @state.setter
    def state(self, state: dict):
+        self._initialized = True
        self._state = state

    @property