adding the selu activation

This commit is contained in:
Itai Caspi authored and Itai Caspi committed 2018-01-22 12:05:43 +02:00
1 parent fff8c8f568
commit 43821c9630
2 files changed
+2

No files matched your search

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