def test_srresnet(self): super_resolution_net = SRResNet( rescale_factor=4, n_filters=64, n_blocks=8) state_dict = model_zoo.load_url(model_urls['srresNet'], progress=False) x = torch.randn(1, 3, 224, 224, requires_grad=True) self.run_model_test(super_resolution_net, train=False, batch_size=1, state_dict=state_dict, input=x, use_gpu=False)
def test_srresnet(self): x = Variable(torch.randn(1, 3, 224, 224).fill_(1.0)) self.exportTest( toC(SRResNet(rescale_factor=4, n_filters=64, n_blocks=8)), toC(x))