Commit 8a8eb34f authored by hubertlu-tw's avatar hubertlu-tw
Browse files

Skip a flaky unit test

parent 975a0e53
......@@ -134,6 +134,7 @@ class TestMLP(unittest.TestCase):
ref_mlp[0].bias.grad.detach().cpu().numpy(),
atol=1e-7, rtol=1e-5)
@skipFlakyTest
def test_no_grad(self):
mlp = MLP(mlp_sizes).cuda()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment