mirror of
https://github.com/gryf/coach.git
synced 2025-12-17 11:10:20 +01:00
adding the selu activation
This commit is contained in:
@@ -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(), \
|
||||
|
||||
Reference in New Issue
Block a user