Avoid loading state dict for automagic for now until I can sort out some issues

2026-04-23 07:49:24 +00:00 · 2025-02-26 17:03:14 -07:00
parent d360e76661
commit dd700f70b3
2 changed files with 63 additions and 0 deletions
--- a/toolkit/optimizers/automagic.py
+++ b/toolkit/optimizers/automagic.py
@@ -322,6 +322,8 @@ class Automagic(torch.optim.Optimizer):
    
    def load_state_dict(self, state_dict, strict=True):
        # load the lr_mask from the state_dict
+        # dont load state dict for now. Has a bug. Need to fix it.
+        return
        idx = 0
        for group in self.param_groups:
            for p in group['params']: