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

adds tests for swish activation

Browse Source
This commit is contained in:
Valentin
2017-11-18 16:54:23 +01:00
parent 942d5c30c1
commit 2f95542184
1 changed files with 12 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/keras_contrib/layers/test_advanced_activations.py
+12
View File
@@ -26,5 +26,17 @@ 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},
input_shape=(2, 3, 4))
@keras_test
def test_swish_trainable():
layer_test(advanced_activations.Swish, kwargs={'beta': 1.0, 'trainable': True},
input_shape=(2, 3, 4))
if __name__ == '__main__':
pytest.main([__file__])