adding the selu activation

This commit is contained in:
Itai Caspi
2018-01-22 12:05:43 +02:00
committed by Itai Caspi
parent fff8c8f568
commit 43821c9630
2 changed files with 2 additions and 0 deletions
@@ -42,6 +42,7 @@ class GeneralNeonNetwork(NeonArchitecture):
'tanh': neon.Tanh(),
'sigmoid': neon.Logistic(),
'elu': neon.Explin(),
'selu': None,
'none': None
}
assert activation_function_string in activation_functions.keys(), \
@@ -45,6 +45,7 @@ class GeneralTensorFlowNetwork(TensorFlowArchitecture):
'tanh': tf.nn.tanh,
'sigmoid': tf.nn.sigmoid,
'elu': tf.nn.elu,
'selu': tf.nn.selu,
'none': None
}
assert activation_function_string in activation_functions.keys(), \