wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-06-27 16:10:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

add lines for pep8

Browse Source
This commit is contained in:
Valentin
2017-12-06 21:36:06 +01:00
parent 7c0d465224
commit 416bbe026c
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/layers/advanced_activations.py
+4 -1
View File
@@ -235,6 +235,9 @@ class SReLU(Layer):
base_config = super(SReLU, self).get_config()
return dict(list(base_config.items()) + list(config.items()))
get_custom_objects().update({'SReLU': SReLU})
class Swish(Layer):
""" Swish (Ramachandranet al., 2017)
@@ -279,4 +282,4 @@ class Swish(Layer):
base_config = super(Swish, self).get_config()
return dict(list(base_config.items()) + list(config.items()))
get_custom_objects().update({'SReLU': SReLU, 'Swish': Swish})
get_custom_objects().update({'Swish': Swish})
tests/keras_contrib/layers/test_advanced_activations.py
+1
View File
@@ -26,6 +26,7 @@ def test_srelu_share():
layer_test(advanced_activations.SReLU, kwargs={'shared_axes': 1},
input_shape=(2, 3, 4))
@keras_test
def test_swish_constant():
layer_test(advanced_activations.Swish, kwargs={'beta': 1.0, 'trainable': False},