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

Fix test to refer to correct field for momentum weights in optimizer

Browse Source
This commit is contained in:
Michael Oliver
2017-07-03 13:28:18 -07:00
parent 13ec4b4174
commit 15133077af
1 changed files with 2 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
+2 -2
View File
@@ -28,7 +28,7 @@ def test_save_and_load_all_weights():
w1value[0, 0:4] = [1, 3, 3, 7]
K.set_value(w1, w1value)
# set optimizer weights
ow1 = m1.optimizer.weights[4] # momentum weights
ow1 = m1.optimizer.weights[3] # momentum weights
ow1value = K.get_value(ow1)
ow1value[0, 0:3] = [4, 2, 0]
K.set_value(ow1, ow1value)
@@ -41,7 +41,7 @@ def test_save_and_load_all_weights():
# 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[4])[0, 0:3], [4, 2, 0])
assert_allclose(K.get_value(m2.optimizer.weights[3])[0, 0:3], [4, 2, 0])
if __name__ == '__main__':