gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

adding the selu activation

Browse Source
This commit is contained in:
Itai Caspi
2018-01-21 12:15:21 +02:00
committed by Itai Caspi
parent fff8c8f568
commit 43821c9630
2 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
architectures/neon_components/general_network.py
View File

@@ -42,6 +42,7 @@ class GeneralNeonNetwork(NeonArchitecture):
'tanh': neon.Tanh(), 'tanh': neon.Tanh(),
'sigmoid': neon.Logistic(), 'sigmoid': neon.Logistic(),
'elu': neon.Explin(), 'elu': neon.Explin(),
'selu': None,
'none': None 'none': None
} }
assert activation_function_string in activation_functions.keys(), \ assert activation_function_string in activation_functions.keys(), \

1
architectures/tensorflow_components/general_network.py
View File

@@ -45,6 +45,7 @@ class GeneralTensorFlowNetwork(TensorFlowArchitecture):
'tanh': tf.nn.tanh, 'tanh': tf.nn.tanh,
'sigmoid': tf.nn.sigmoid, 'sigmoid': tf.nn.sigmoid,
'elu': tf.nn.elu, 'elu': tf.nn.elu,
'selu': tf.nn.selu,
'none': None 'none': None
} }
assert activation_function_string in activation_functions.keys(), \ assert activation_function_string in activation_functions.keys(), \