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
@@ -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(), \