This commit is contained in:
Nataniel Ruiz
2019-12-25 15:23:45 -04:00
parent 16857d1569
commit 79fa680c1b
3 changed files with 5 additions and 4 deletions
+1 -1
View File
@@ -241,7 +241,7 @@ class Pix2PixHDModel(BaseModel):
else:
input_concat = input_label
input_adv = torch.clamp(input_concat + perturb * 1000, min=-1, max=1)
input_adv = torch.clamp(input_concat + perturb, min=-1, max=1)
with torch.no_grad():
fake_image = self.netG.forward(input_adv)