Skip to content

Commit

Permalink
enable test_mnist
Browse files Browse the repository at this point in the history
  • Loading branch information
KsenijaS committed Oct 1, 2020
1 parent 78c1cb4 commit d4ba1f5
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion test/onnx/test_models.py
Expand Up @@ -105,7 +105,6 @@ def test_alexnet(self):
)
self.exportTest(toC(alexnet()), toC(x))

@disableScriptTest()
def test_mnist(self):
x = Variable(torch.randn(BATCH_SIZE, 1, 28, 28).fill_(1.0))
self.exportTest(toC(MNIST()), toC(x))
Expand Down

0 comments on commit d4ba1f5

Please sign in to comment.