Skip to content

Commit 2263eb5

Browse files
committed
update docs for save_npz_dict load_npz_dict
1 parent d7b65f4 commit 2263eb5

File tree

2 files changed

+15
-5
lines changed

2 files changed

+15
-5
lines changed

docs/modules/files.rst

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,9 @@ sake of cross-platform.
4242
load_wmt_en_fr_dataset
4343

4444
save_npz
45+
save_npz_dict
4546
load_npz
47+
load_npz_dict
4648
assign_params
4749
load_and_assign_npz
4850

@@ -92,14 +94,22 @@ English-to-French translation data from the WMT'15 Website
9294
Load and save network
9395
----------------------
9496

95-
Save network as .npz
96-
^^^^^^^^^^^^^^^^^^^^^^^^
97+
Save network into list
98+
^^^^^^^^^^^^^^^^^^^^^^^^^^
9799
.. autofunction:: save_npz
98100

99-
Load network from .npz
100-
^^^^^^^^^^^^^^^^^^^^^^^^^^
101+
Save network into dict
102+
^^^^^^^^^^^^^^^^^^^^^^^^
103+
.. autofunction:: save_npz_dict
104+
105+
Load network from save_npz
106+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
101107
.. autofunction:: load_npz
102108

109+
Load network from save_npz_dict
110+
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
111+
.. autofunction:: load_npz_dict
112+
103113
Assign parameters to network
104114
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
105115
.. autofunction:: assign_params

tensorlayer/files.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -678,7 +678,7 @@ def assign_params(sess, params, network):
678678
>>> network.print_params()
679679
...
680680
... Next time, load and assign your network as follow:
681-
>>> sess.run(tf.initialize_all_variables()) # re-initialize, then save and assign
681+
>>> tl.layers.initialize_global_variables(sess)
682682
>>> load_params = tl.files.load_npz(name='model_test.npz')
683683
>>> tl.files.assign_params(sess, load_params, network)
684684
>>> network.print_params()

0 commit comments

Comments
 (0)