[DEL] Remove inputs from state attributes

f-dangel · Nov 3, 2023 · 1d75e9b · 1d75e9b
1 parent d97d316
commit 1d75e9b
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/singd/optim/optimizer.py b/singd/optim/optimizer.py
@@ -848,7 +848,6 @@ def set_current_grad_scale(self, grad_scale: float):
         "m_Cs",
         "H_Ks",
         "H_Cs",
-        "inputs",
     ]
 
     def state_dict(self) -> Dict[str, Any]: