Merge branch 'fail-counter' into fail-counter-v2

facebookresearch · Dec 10, 2024 · 8f19d2f · 8f19d2f
2 parents 196a781 + d53ffc6
commit 8f19d2f
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/distributed_shampoo/shampoo_types.py b/distributed_shampoo/shampoo_types.py
@@ -8,7 +8,7 @@
 """
 
 import enum
-from dataclasses import dataclass
+from dataclasses import dataclass, field
 
 import torch
 
@@ -109,7 +109,9 @@ class ShampooPreconditionerConfig(PreconditionerConfig):
 
     """
 
-    amortized_computation_config: RootInvConfig = DefaultEigenConfig
+    amortized_computation_config: RootInvConfig = field(
+        default_factory=lambda: DefaultEigenConfig
+    )
 
 
 DefaultShampooConfig = ShampooPreconditionerConfig()
@@ -124,7 +126,9 @@ class EigenvalueCorrectedShampooPreconditionerConfig(PreconditionerConfig):
 
     """
 
-    amortized_computation_config: EigenvectorConfig = DefaultEighConfig
+    amortized_computation_config: EigenvectorConfig = field(
+        default_factory=lambda: DefaultEighConfig
+    )
 
 
 DefaultEigenvalueCorrectedShampooConfig = (