File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments