Skip to content

Commit 17c29e8

Browse files
committed
update docs save / load npz
1 parent 07d16a8 commit 17c29e8

1 file changed

Lines changed: 7 additions & 6 deletions

File tree

docs/modules/files.rst

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -9,21 +9,22 @@ sake of cross-platform.
99

1010
.. code-block:: python
1111
12-
# save model as .ckpt
12+
## save model as .ckpt
1313
saver = tf.train.Saver()
1414
save_path = saver.save(sess, "model.ckpt")
1515
# restore model from .ckpt
1616
saver = tf.train.Saver()
1717
saver.restore(sess, "model.ckpt")
1818
19-
# save model as .npz
19+
## save model as .npz
2020
tl.files.save_npz(network.all_params , name='model.npz')
21-
22-
# restore model from .npz
23-
load_params = tl.files.load_npz(path='', name='model.npz')
21+
# restore model from .npz (method 1)
22+
load_params = tl.files.load_npz(name='model.npz')
2423
tl.files.assign_params(sess, load_params, network)
24+
# restore model from .npz (method 2)
25+
tl.files.load_and_assign_npz(sess=sess, name='model.npz', network=network)
2526
26-
# you can assign the pre-trained parameters as follow
27+
## you can assign the pre-trained parameters as follow
2728
# 1st parameter
2829
tl.files.assign_params(sess, [load_params[0]], network)
2930
# the first three parameters

0 commit comments

Comments
 (0)