keras.layers.ReLU added (apache#1530)

sergei-mironov · Aug 8, 2018 · 21b3c07 · 21b3c07
1 parent fd2cd0c
commit 21b3c07
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/nnvm/python/nnvm/frontend/keras.py b/nnvm/python/nnvm/frontend/keras.py
@@ -74,7 +74,9 @@ def _convert_activation(insym, keras_layer, _):
 
 def _convert_advanced_activation(insym, keras_layer, symtab):
     act_type = type(keras_layer).__name__
-    if act_type == 'LeakyReLU':
+    if act_type == 'ReLU':
+        return _sym.relu(insym)
+    elif act_type == 'LeakyReLU':
         return _sym.leaky_relu(insym, alpha=keras_layer.alpha)
     elif act_type == 'ELU':
         alpha = keras_layer.alpha if hasattr(keras_layer, "alpha") else 1
@@ -382,6 +384,7 @@ def _default_skip(insym, keras_layer, _): # pylint: disable=unused-argument
 _convert_map = {
     'Dense'                    : _convert_dense,
     'Activation'               : _convert_activation,
+    'ReLU'                     : _convert_advanced_activation,
     'LeakyReLU'                : _convert_advanced_activation,
     'PReLU'                    : _convert_advanced_activation,
     'ELU'                      : _convert_advanced_activation,

diff --git a/nnvm/tests/python/frontend/keras/test_forward.py b/nnvm/tests/python/frontend/keras/test_forward.py
@@ -138,6 +138,7 @@ def test_forward_activations():
     weights = np.random.rand(1, 32, 32, 3)
     act_funcs = [keras.layers.Activation('softmax'),
                  keras.layers.Activation('softplus'),
+                 keras.layers.ReLU(),
                  keras.layers.LeakyReLU(alpha=0.3),
                  keras.layers.PReLU(weights=weights, alpha_initializer="zero"),
                  keras.layers.ELU(alpha=0.5),