CalvinBackup/disrupting-deepfakes
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

next

Browse Source
This commit is contained in:
Nataniel Ruiz
2019-12-26 12:52:43 -04:00
parent 0df250a64c
commit d3dc2c60fd
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cyclegan/test.py
+3 -3
View File
@@ -68,9 +68,9 @@ if __name__ == '__main__':
model.set_input(data) # unpack data from data loader model.set_input(data) # unpack data from data loader
with torch.no_grad(): with torch.no_grad():
model.forward_noattack() model.forward_noattack()
if i == 0: # if i == 0:
input_adv, perturb = model.attack() # input_adv, perturb = model.attack()
# input_adv, perturb = model.attack() input_adv, perturb = model.attack()
with torch.no_grad(): with torch.no_grad():
model.forward_attack(perturb) model.forward_attack(perturb)
model.compute_visuals() model.compute_visuals()