From 6dbcf6f93b2534a5f5728558be5cbfb00fd10da5 Mon Sep 17 00:00:00 2001
From: Nataniel Ruiz <nruiz@Nataniels-MacBook-Pro.local>
Date: Mon, 30 Dec 2019 13:53:13 -0400
Subject: [PATCH] next

---
 stargan/attacks.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/stargan/attacks.py b/stargan/attacks.py
index fa2acf0..8aa59db 100644
--- a/stargan/attacks.py
+++ b/stargan/attacks.py
@@ -46,7 +46,7 @@ class LinfPGDAttack(object):
 
         self.model.zero_grad()
 
-        return X, eta
+        return X, X - X_nat
 
 def clip_tensor(X, Y, Z):
     # Clip X with Y min and Z max