Skip to content

Commit 0e4228b

Browse files
fix test_weight_prepack.py ut (#723)
1 parent 935bd77 commit 0e4228b

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
@@ -321,7 +321,7 @@ def _test_imagenet_model(self, model):
321321
ipex_optimizer.zero_grad()
322322
loss2.backward()
323323
ipex_optimizer.step()
324-
self.assertEqual(y1, y2.float(), rtol=5e-2, atol=1e-2)
324+
self.assertEqual(y1, y2.float(), rtol=5e-2, atol=5e-2)
325325

326326

327327
@skipIfNoTorchVision

0 commit comments

Comments
 (0)