Update Time_TravelRephotography/op/fused_act.py

Time_TravelRephotography/op/fused_act.py

@@ -71,10 +71,9 @@ class FusedLeakyReLUFunction(Function):
 
 
 class FusedLeakyReLU(nn.Module):
-    def __init__(self, channel, negative_slope=0.2, scale=2 ** 0.5):
+    def __init__(self, negative_slope=0.2, scale=2 ** 0.5):
         super().__init__()
 
-        self.bias = nn.Parameter(torch.zeros(channel))
         self.negative_slope = negative_slope
         self.scale = scale
 
@@ -82,6 +81,6 @@ class FusedLeakyReLU(nn.Module):
         return fused_leaky_relu(input, self.bias, self.negative_slope, self.scale)
 
 
-def fused_leaky_relu(input):
-    return FusedLeakyReLU.apply(input)
+def fused_leaky_relu(input, bias):
+    return FusedLeakyReLU.apply(input, bias)