def test_alexnet(self):
alexnet = AlexNet()
state_dict = model_zoo.load_url(model_urls['alexnet'], progress=False)
self.run_model_test(alexnet,
train=False,
batch_size=BATCH_SIZE,
state_dict=state_dict)
def test_alexnet(self):
x = Variable(torch.randn(BATCH_SIZE, 3, 224, 224).fill_(1.0))
self.exportTest(toC(AlexNet()), toC(x))