From 0f5f9abbd2ea1806ef5a81b4f25cb65a17183f01 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Mon, 23 Dec 2019 16:49:26 -0400 Subject: [PATCH] next --- stargan/attacks.py | 2 +- stargan/model.py | 4 ++-- stargan/solver.py | 4 ++-- 3 files changed, 5 insertions(+), 5 deletions(-) diff --git a/stargan/attacks.py b/stargan/attacks.py index 80424c7..5c8b88c 100644 --- a/stargan/attacks.py +++ b/stargan/attacks.py @@ -29,7 +29,7 @@ class LinfPGDAttack(object): output, feats = self.model(X, c_trg) if self.feat: - print('self.feat ', self.feat) + # print('self.feat ', self.feat) output = feats[self.feat] y = np.zeros(output.shape) y = torch.FloatTensor(y).to(self.device) diff --git a/stargan/model.py b/stargan/model.py index beb7d82..a8e91e8 100644 --- a/stargan/model.py +++ b/stargan/model.py @@ -66,10 +66,10 @@ class Generator(nn.Module): # Get intermediate feature maps for layer in self.main: - print('test') + # print('test') x = layer(x) feature_maps.append(x) - print(len(feature_maps)) + # print(len(feature_maps)) return x, feature_maps diff --git a/stargan/solver.py b/stargan/solver.py index 8db1639..1d63e65 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -587,7 +587,7 @@ class Solver(object): n_samples = 0 # 11 layers - layer_num = 0 + layer_num_orig = 0 for i, (x_real, c_org) in enumerate(data_loader): # Black image @@ -598,7 +598,7 @@ class Solver(object): x_real = x_real.to(self.device) c_trg_list = self.create_labels(c_org, self.c_dim, self.dataset, self.selected_attrs) - layer_num = (layer_num + 1) * 3 - 1 + layer_num = (layer_num_orig + 1) * 3 - 1 pgd_attack = attacks.LinfPGDAttack(model=self.G, device=self.device, feat=layer_num) # Translate images.