From 7935838a62fca09b0b52c9fca0fa154ef1313f16 Mon Sep 17 00:00:00 2001 From: Tianming Xu Date: Tue, 12 Jan 2021 18:09:48 +0800 Subject: [PATCH] add default value for leaky relu alpha (#7259) --- python/tvm/relay/op/nn/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/relay/op/nn/nn.py b/python/tvm/relay/op/nn/nn.py index fef82e7c1fd3..562cee5f53bb 100644 --- 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.