tensorlayer layer加载model.npz

来源:互联网 发布:铁幕演说知乎 编辑:程序博客网 时间:2024/06/06 20:52

tensorlayer的模型可以通过save_npz(network.all_params , name='model.npz')保存成文件

下次使用可以通过load_npz(npzPath)加载保存的数据

通过代码赋值到layer中的all_params

   if os.path.exists(npzPath):
       npz = fl.load_npz(npzPath)
       var_list = npz.tolist()
       for k,val in enumerate(var_list):
           w=network.all_params[k].assign(val)
           sess.run(w);


0 0
原创粉丝点击