Skip to content

Commit 77068a0

Browse files
change rtol to 1e-4 when compare model with different optimize level (#138)
1 parent c2edd8f commit 77068a0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/cpu/test_weight_prepack.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,7 @@ def _test_imagenet_model(self, model):
294294
loss2.backward()
295295
ipex_optimizer2.step()
296296
self.assertEqual(y, y1)
297-
self.assertEqual(y1, y2, rtol=1e-5, atol=1e-3)
297+
self.assertEqual(y1, y2, rtol=1e-4, atol=1e-3)
298298
self.assertEqual(loss, loss1)
299299
self.assertEqual(loss1, loss2, rtol=1e-5, atol=1e-3)
300300

0 commit comments

Comments
 (0)