Tensorflow 保存和加载model
保存model的代码:
import tensorflow as tf
save_model_path = "/save_model" # 保存的文件夹路径
network.save(save_model_path);
保存的目录格式如下,期中saved_model.pb是主要的文件:
save_model
-- assets
-- variables
-- variables.data-00000-of-00001
-- variables.index
-- saved_model.pb
加载model如下:
