Update audio_diffusion_attacks_forhf/src/balancer.py

audio_diffusion_attacks_forhf/src/balancer.py

@@ -92,7 +92,8 @@ class Balancer:
             # Compute partial derivative of the less with respect to the input.
             #Andy added:
             loss.requires_grad = True
-            grad, = autograd.grad(loss, [input], retain_graph=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)
             if self.per_batch_item:
                 # We do not average the gradient over the batch dimension.
                 dims = tuple(range(1, grad.dim()))