From bf72ae1dcb1d7bc064c4496cd327b76b3cb85db1 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Mon, 30 Dec 2019 14:17:33 -0400 Subject: [PATCH] next --- stargan/attacks.py | 2 +- stargan/solver.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/stargan/attacks.py b/stargan/attacks.py index b39b885..0f76461 100644 --- a/stargan/attacks.py +++ b/stargan/attacks.py @@ -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 3e87540..aafaa44 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -594,6 +594,7 @@ class Solver(object): n_samples = 0 for i, (x_real, c_org) in enumerate(data_loader): + self.restore_model(self.test_iters) # Black image black = np.zeros((1,3,256,256)) black = torch.FloatTensor(black).to(self.device)