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

Merge pull request #162 from farizrahman4u/save_load_utils_tf_only

Browse Source 
save_load_utils_test.py becomes tensorflow only to fix test failure until #157 is resolved
This commit is contained in:
Andrew Hundt
2017-10-15 14:23:42 -04:00
committed by GitHub
parent 99fe5e3b3a 70e68c11f5
commit 7dc329a5a3
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.