Update audio_diffusion_attacks_forhf/src/balancer.py

audio_diffusion_attacks_forhf/src/balancer.py

@@ -90,10 +90,7 @@ class Balancer:
         grads = {}
         for name, loss in losses.items():
             # Compute partial derivative of the less with respect to the input.
-            #Andy added:
-            loss.requires_grad = True
-            #Andy edited--CHECK WITH WILLIAM: grad, = autograd.grad(loss, [input], retain_graph=True)
-            grad, = autograd.grad(loss, [input], retain_graph=True, allow_unused=True)
+            grad, = autograd.grad(loss, [input], retain_graph=True)
             if self.per_batch_item:
                 # We do not average the gradient over the batch dimension.
                 dims = tuple(range(1, grad.dim()))