feng2022 commited on
Commit
2e0063b
1 Parent(s): 71ff851

Update Time_TravelRephotography/op/fused_act.py

Browse files
Time_TravelRephotography/op/fused_act.py CHANGED
@@ -82,6 +82,6 @@ class FusedLeakyReLU(nn.Module):
82
  return fused_leaky_relu(input, self.bias, self.negative_slope, self.scale)
83
 
84
 
85
- def fused_leaky_relu(input, bias, negative_slope=0.2, scale=2 ** 0.5):
86
- return FusedLeakyReLU.apply(input, bias, negative_slope, scale)
87
 
 
82
  return fused_leaky_relu(input, self.bias, self.negative_slope, self.scale)
83
 
84
 
85
+ def fused_leaky_relu(input, bias):
86
+ return FusedLeakyReLU.apply(input, bias)
87