Update functions.py

2025-03-17 15:45:21 +09:00
parent 40605a549a
commit 13b380b86b
1 changed files with 1 additions and 1 deletions
@@ -46,7 +46,7 @@ class InPlaceABN(autograd.Function):
        ctx.save_for_backward(x_norm, var, weight, bias)
        ctx.mark_non_differentiable(running_mean, running_var)
        
-        return x_norm
+        return x_norm, running_mean, running_var
    
    @staticmethod
    @once_differentiable