default adopt atan2 for export given experiments im seeing

lucidrains · Nov 22, 2024 · 661aed2 · 661aed2
1 parent 0177579
commit 661aed2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/adam_atan2_pytorch/__init__.py b/adam_atan2_pytorch/__init__.py
@@ -1,4 +1,5 @@
 from adam_atan2_pytorch.adam_atan2 import AdamAtan2
-from adam_atan2_pytorch.adopt import Adopt
+from adam_atan2_pytorch.adopt_atan2 import AdoptAtan2
 
 Adam = AdamAtan2
+Adopt = AdoptAtan2
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "adam-atan2-pytorch"
-version = "0.1.7"
+version = "0.1.8"
 description = "Adam-atan2 for Pytorch"
 authors = [
     { name = "Phil Wang", email = "[email protected]" }