From a3c3a9430594b3109b314d02b81f91d9026c5f69 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Wed, 5 Sep 2018 09:21:01 +0200 Subject: [PATCH] Update. --- hallu.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/hallu.py b/hallu.py index 6b0b303..7da66a6 100755 --- a/hallu.py +++ b/hallu.py @@ -12,7 +12,7 @@ class MultiScaleEdgeEnergy(torch.nn.Module): super(MultiScaleEdgeEnergy, self).__init__() k = torch.exp(- torch.tensor([[-2., -1., 0., 1., 2.]])**2 / 2) k = (k.t() @ k).view(1, 1, 5, 5) - self.register_buffer('gaussian_5x5', k / k.sum()) + self.gaussian_5x5 = torch.nn.Parameter(k / k.sum()).requires_grad_(False) def forward(self, x): u = x.view(-1, 1, x.size(2), x.size(3)) @@ -43,7 +43,7 @@ for l in [ 5, 7, 12, 17, 21, 28 ]: ref_output = model(ref_input).detach() for n in range(5): - input = ref_input.new_empty(ref_input.size()).uniform_(-0.01, 0.01).requires_grad_() + input = torch.empty_like(ref_input).uniform_(-0.01, 0.01).requires_grad_() optimizer = torch.optim.Adam( [ input ], lr = 1e-2) for k in range(1000): output = model(input) -- 2.20.1