From 1102475fe3718fcd7d06e74659b0994b85c6f7d6 Mon Sep 17 00:00:00 2001
From: Nataniel Ruiz <nruiz@Nataniels-MacBook-Pro.local>
Date: Wed, 25 Dec 2019 20:35:23 -0400
Subject: [PATCH] next

---
 cyclegan/models/test_model.py | 2 ++
 cyclegan/test.py              | 3 ++-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/cyclegan/models/test_model.py b/cyclegan/models/test_model.py
index 7b5aaea..65145f5 100644
--- a/cyclegan/models/test_model.py
+++ b/cyclegan/models/test_model.py
@@ -78,6 +78,8 @@ class TestModel(BaseModel):
         black = np.zeros((1, 3, image.size(2), image.size(3)))
         black = torch.FloatTensor(black).cuda()
         input_adv, perturb = pgd_attack.perturb(image, black)      
+
+        black = None
         
         return input_adv, perturb
 
diff --git a/cyclegan/test.py b/cyclegan/test.py
index 916d5bb..33cd4fa 100644
--- a/cyclegan/test.py
+++ b/cyclegan/test.py
@@ -88,7 +88,8 @@ if __name__ == '__main__':
         model.fake = None
         model.real = None
         model.fake_noattack = None
-        
+        data = None
+
     webpage.save()  # save the HTML