Bag fix in optim.recover_state()

alex-snd · Sep 28, 2022 · 34097f3 · 34097f3
1 parent 72fe8c5
commit 34097f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/trecover/train/collab/optim.py b/src/trecover/train/collab/optim.py
@@ -597,7 +597,7 @@ def recover_state(self, sync: bool = False) -> None:
         if sync:
             while not self.params_are_finite:
                 self.restore_from_backup()
-                self.sync_state()
+                self.sync_state(sync=True)
         else:
             self.restore_from_backup()