[FBcode->GH] Add check_stride=False (#3920)

pytorch · May 26, 2021 · 48de528 · 48de528
1 parent f584309
commit 48de528
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_transforms.py b/test/test_transforms.py
@@ -834,7 +834,7 @@ def test_accimage_pil_to_tensor(self):
         output = trans(accimage.Image(GRACE_HOPPER))
 
         self.assertEqual(expected_output.size(), output.size())
-        torch.testing.assert_close(output, expected_output)
+        torch.testing.assert_close(output, expected_output, check_stride=False)
 
     @unittest.skipIf(accimage is None, 'accimage not available')
     def test_accimage_resize(self):