Update audio_diffusion_attacks_forhf/src/balancer.py

audio_diffusion_attacks_forhf/src/balancer.py

@@ -90,6 +90,8 @@ 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
             grad, = autograd.grad(loss, [input], retain_graph=True)
             if self.per_batch_item:
                 # We do not average the gradient over the batch dimension.