feng2022 commited on
Commit
71ff851
1 Parent(s): 5d42f5f

Update Time_TravelRephotography/op/fused_act.py

Browse files
Time_TravelRephotography/op/fused_act.py CHANGED
@@ -83,5 +83,5 @@ class FusedLeakyReLU(nn.Module):
83
 
84
 
85
  def fused_leaky_relu(input, bias, negative_slope=0.2, scale=2 ** 0.5):
86
- return FusedLeakyReLUFunction.apply(input, bias, negative_slope, scale)
87
 
 
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