From 10836dcd6692ea87d44fbd8703ad4e336af60f24 Mon Sep 17 00:00:00 2001 From: Nataniel Ruiz Date: Tue, 24 Dec 2019 19:59:51 -0400 Subject: [PATCH] next --- stargan/attacks.py | 2 +- stargan/solver.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/stargan/attacks.py b/stargan/attacks.py index 49a29ee..fa2acf0 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=20, a=0.01, feat = None): + def __init__(self, model=None, device=None, epsilon=0.05, k=1, a=0.05, feat = None): self.model = model self.epsilon = epsilon self.k = k diff --git a/stargan/solver.py b/stargan/solver.py index 4739762..ee0c554 100644 --- a/stargan/solver.py +++ b/stargan/solver.py @@ -594,10 +594,10 @@ class Solver(object): for i, (x_real, c_org) in enumerate(data_loader): # Black image - black = np.zeros((1,3,256,256)) - black = torch.FloatTensor(black).to(self.device) + # black = np.zeros((1,3,256,256)) + # black = torch.FloatTensor(black).to(self.device) - # black = torch.FloatTensor(torch.rand((1,3,256,256))).to(self.device) + black = torch.FloatTensor(torch.rand((1,3,256,256))).to(self.device) # Prepare input images and target domain labels. x_real = x_real.to(self.device)