Skip to content

Commit 417bb0f

Browse files
committed
chnaged wd to 0.25 for tolerance and removed error msg override
Signed-off-by: mikail <[email protected]>
1 parent 50f613b commit 417bb0f

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/test_orthogonalized_optimizer.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -191,7 +191,7 @@ def test_use_syrk_match_without_syrk(self) -> None:
191191
def test_use_independent_wd(self) -> None:
192192
"""Test that use_independent_wd properly decouples weight decay from learning rate."""
193193
shape = (32, 32)
194-
weight_decay = 0.1
194+
weight_decay = 0.25
195195

196196
# Test with independent weight decay: with lr=0, weight decay should still be applied
197197
# With lr=0, no gradient update occurs, so param should be exactly (1-wd)*param
@@ -215,7 +215,6 @@ def test_use_independent_wd(self) -> None:
215215
expected_param,
216216
atol=0,
217217
rtol=0,
218-
msg=f"Expected param to be exactly (1-{weight_decay})*original_param with independent weight decay and lr=0",
219218
)
220219

221220

0 commit comments

Comments
 (0)