Update methods.py

methods.py

@@ -161,6 +161,8 @@ def averaged_transforms(guided, ssl_model, mixed_images, blur_output):
     grads2 = []
 
     for xbar_image in mixed_images[1:]:  
+
+        ssl_model.zero_grad()
         input_image1 = mixed_images[0].clone().requires_grad_()
         input_image2 = xbar_image.clone().requires_grad_()
 
@@ -241,6 +243,8 @@ def smooth_grad(guided, ssl_model, img1, img2, blur_output, steps = 50):
     grads2 = []
 
     for n1, n2 in zip(noise_images1, noise_images2):  
+
+        ssl_model.zero_grad()
         input_image1 = n1.clone().requires_grad_()
         input_image2 = n2.clone().requires_grad_()