def test_mnist(self):
model = MNIST()
input = Variable(torch.randn(BATCH_SIZE, 1, 28, 28))
state_dict = None
# TODO: test with state_dict
self.run_model_test(model,
train=False,
input=input,
batch_size=BATCH_SIZE,
state_dict=state_dict)
def test_mnist(self):
x = Variable(torch.randn(BATCH_SIZE, 1, 28, 28).fill_(1.0))
self.exportTest(toC(MNIST()), toC(x))
