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 #161 from farizrahman4u/rm_test_file

Browse Source 
save_load_utils_test.py removes 'model.h5'
This commit is contained in:
Andrew Hundt
2017-10-13 09:57:28 -04:00
committed by GitHub
parent a92268ffcb 6ec66fec74
commit de768b6048
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/keras_contrib/utils/save_load_utils_test.py
+4 -2
View File
@@ -1,4 +1,5 @@
import pytest
import os
from keras import backend as K
from keras.layers import Input, Dense
from keras.models import Model
@@ -33,15 +34,16 @@ def test_save_and_load_all_weights():
ow1value[0, 0:3] = [4, 2, 0]
K.set_value(ow1, ow1value)
# save all weights
save_all_weights(m1, "model.h5")
save_all_weights(m1, 'model.h5')
# new model
m2 = make_model()
# load all weights
load_all_weights(m2, "model.h5")
load_all_weights(m2, 'model.h5')
# check weights
assert_allclose(K.get_value(m2.layers[1].kernel)[0, 0:4], [1, 3, 3, 7])
# check optimizer weights
assert_allclose(K.get_value(m2.optimizer.weights[3])[0, 0:3], [4, 2, 0])
os.remove('model.h5')
if __name__ == '__main__':