add default value for leaky relu alpha (apache#7259)

electriclilies · Feb 18, 2021 · 02c5aa5 · 02c5aa5
1 parent 3e30a60
commit 02c5aa5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/relay/op/nn/nn.py b/python/tvm/relay/op/nn/nn.py
@@ -1488,7 +1488,7 @@ def relu(data):
     return _make.relu(data)
 
 
-def leaky_relu(data, alpha):
+def leaky_relu(data, alpha=0.01):
     """This operator takes data as input and does Leaky version
     of a Rectified Linear Unit.