Skip to content

Commit cf34483

Browse files
author
Seppo Enarvi
committed
Increased test_ema_resume tolerance
1 parent 822231f commit cf34483

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/tests_pytorch/callbacks/test_weight_averaging.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -235,7 +235,7 @@ def test_ema_resume(tmp_path, crash_on_epoch):
235235
model2 = _train_and_resume(model2, dataset, tmp_path)
236236

237237
for param1, param2 in zip(model1.parameters(), model2.parameters()):
238-
assert torch.allclose(param1, param2, atol=0.001)
238+
assert torch.allclose(param1, param2, atol=0.005)
239239

240240

241241
@RunIf(skip_windows=True)

0 commit comments

Comments
 (0)