Skip to content

Commit 110ea77

Browse files
committed
works fine for me
1 parent bbbecbe commit 110ea77

File tree

3 files changed

+6
-0
lines changed

3 files changed

+6
-0
lines changed

adam_atan2_pytorch/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1 +1,3 @@
11
from adam_atan2_pytorch.adam_atan2 import AdamAtan2
2+
3+
Adam = AdamAtan2

adam_atan2_pytorch/adam_atan2_with_wasserstein_reg.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,3 +137,5 @@ def step(
137137
state['steps'] = steps
138138

139139
return loss
140+
141+
Adam = AdamAtan2

adam_atan2_pytorch/foreach.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,3 +168,5 @@ def step(
168168
torch._foreach_add_(params, updates, alpha = -lr * a)
169169

170170
return loss
171+
172+
Adam = AdamAtan2

0 commit comments

Comments
 (0)