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

save_load_utils_test.py make tensorflow only

Browse Source 
Making this test tensorflow only until https://github.com/farizrahman4u/keras-contrib/issues/157 is resolved.
This commit is contained in:
Andrew Hundt
2017-10-15 13:28:43 -04:00
committed by GitHub
parent 99fe5e3b3a
commit 70e68c11f5
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/keras_contrib/utils/save_load_utils_test.py
+3
View File
@@ -4,10 +4,13 @@ from keras import backend as K
from keras.layers import Input, Dense
from keras.models import Model
from numpy.testing import assert_allclose
from keras.utils.test_utils import keras_test
from keras_contrib.utils.save_load_utils import save_all_weights, load_all_weights
@pytest.mark.skipif(K.backend() != 'tensorflow', reason='save_all_weights and load_all_weights only supported on TensorFlow')
@keras_test
def test_save_and_load_all_weights():
'''
Test save_all_weights and load_all_weights. Save and load optimizer and model weights but not configuration.