Update Time_TravelRephotography/torch_utils/ops/bias_act.py

Time_TravelRephotography/torch_utils/ops/bias_act.py

@@ -89,9 +89,9 @@ def bias_act(x, b=None, dim=1, act='linear', alpha=None, gain=None, clamp=None,
     return _bias_act_ref(x=x, b=b, dim=dim, act=act, alpha=alpha, gain=gain, clamp=clamp)
 
 class bias_act_relu(nn.Module):
-    def __init__(self, dim, negative_slope=0.2, scale=2 ** 0.5):
+    def __init__(self, bias, negative_slope=0.2, scale=2 ** 0.5):
         super().__init__()
-        self.dim = nn.Parameter(torch.zeros(dim))
+        self.bias = nn.Parameter(torch.zeros(bias))
         self.negative_slope = negative_slope
         self.scale = scale