Update Time_TravelRephotography/op/fused_act.py

Time_TravelRephotography/op/fused_act.py

@@ -74,7 +74,7 @@ class FusedLeakyReLU(nn.Module):
     def __init__(self, channel, negative_slope=0.2, scale=2 ** 0.5):
         super().__init__()
         channel = torch.tensor(channel)
-        self.bias = nn.Parameter(torch.zeros(tensor.size(channel)))
+        self.bias = nn.Parameter(torch.zeros(channel.shape))
         self.negative_slope = negative_slope
         self.scale = scale