From 0737326238fda9fc7d73ea4905759d5e4cc7cafc Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Mon, 30 Dec 2019 14:13:43 -0400 Subject: [PATCH] next --- stargan/attacks.py | 4 ++-- stargan/solver.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/stargan/attacks.py b/stargan/attacks.py index 8aa59db..b39b885 100644 --- a/stargan/attacks.py +++ b/stargan/attacks.py @@ -7,7 +7,7 @@ import torch import torch.nn as nn class LinfPGDAttack(object): - def __init__(self, model=None, device=None, epsilon=0.05, k=1, a=0.05, feat = None): + def __init__(self, model=None, device=None, epsilon=0.05, k=1, a=0.01, feat = None): self.model = model self.epsilon = epsilon self.k = k @@ -24,7 +24,7 @@ class LinfPGDAttack(object): X = X_nat.clone().detach_() for i in range(self.k): - # print(i) + print(i) X.requires_grad = True output, feats = self.model(X, c_trg) diff --git a/stargan/solver.py b/stargan/solver.py index 6458439..3e87540 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -611,7 +611,6 @@ class Solver(object): # x_advs = [] if i == 0: - print(i) _, perturb = pgd_attack.perturb(x_real, black, c_trg_list[0]) # for idx, c_trg in enumerate(c_trg_list): # x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg) @@ -624,7 +623,8 @@ class Solver(object): # Attack # x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg) # _, perturb = x_advs[idx] - x_adv = x_real + perturb + # x_adv = x_real + perturb + x_adv = torch.clamp(x_real + perturb, min=-1, max=1) # x_adv = self.blur_tensor(x_adv) # Metrics