From b8e12cb42e1b234942f10fd6c83fdf9fa2c50031 Mon Sep 17 00:00:00 2001 From: Baoyuan Qi Date: Mon, 20 Dec 2021 15:39:13 +0800 Subject: [PATCH] instance support for act layer instance support for act layer --- deepctr/layers/activation.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/deepctr/layers/activation.py b/deepctr/layers/activation.py index 5e55945f..e5b77938 100644 --- a/deepctr/layers/activation.py +++ b/deepctr/layers/activation.py @@ -70,6 +70,8 @@ def activation_layer(activation): act_layer = tf.keras.layers.Activation(activation) elif issubclass(activation, Layer): act_layer = activation() + elif isinstance(activation, Layer): + act_layer = activation else: raise ValueError( "Invalid activation,found %s.You should use a str or a Activation Layer Class." % (activation))