Update Time_TravelRephotography/torch_utils/ops/bias_act.py

Time_TravelRephotography/torch_utils/ops/bias_act.py

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