Skip to content

Commit 3c450db

Browse files
committed
address guo's comment
1 parent dd8e778 commit 3c450db

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

tensorlayer/files.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1160,7 +1160,7 @@ def save_npz(save_list=None, name='model.npz', sess=None):
11601160
11611161
References
11621162
----------
1163-
- `Saving dictionary using numpy <http://stackoverflow.com/questions/22315595/saving-dictionary-of-header-information-using-numpy-savez>`__
1163+
`Saving dictionary using numpy <http://stackoverflow.com/questions/22315595/saving-dictionary-of-header-information-using-numpy-savez>`__
11641164
11651165
"""
11661166
if save_list is None:
@@ -1171,8 +1171,7 @@ def save_npz(save_list=None, name='model.npz', sess=None):
11711171
save_list_var = sess.run(save_list)
11721172
else:
11731173
try:
1174-
for _k, v in enumerate(save_list):
1175-
save_list_var.append(v.eval())
1174+
save_list_var.extend([v.eval() for v in save_list])
11761175
except Exception:
11771176
logging.info(" Fail to save model, Hint: pass the session into this function, tl.files.save_npz(network.all_params, name='model.npz', sess=sess)")
11781177
np.savez(name, params=save_list_var)

0 commit comments

Comments
 (0)