next

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
 

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