Import AdamO and SGDO by default

kinit-sk · Jan 17, 2025 · 4930695 · 4930695
1 parent 7a993bc
commit 4930695
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/overshoot/__init__.py b/overshoot/__init__.py
@@ -0,0 +1,2 @@
+from .adamw_overshoot_delayed import AdamO
+from .sgd_overshoot import SGDO