tensorflow-CNN保存模型

init= tf.initialize_all_variables()

saver = tf.train.Saver()

# 用 saver 将所有的 variable 保存到定义的路径

withtf.Session()assess:

sess.run(init)

save_path = saver.save(sess,"./save_cnn_net.ckpt")

print("Save to path: ", save_path)

将运行完的CNN模型保存到save_cnn_net.ckpt中

其中文件名要是“./*****”,不然会报错不存在现象,错误如下

.ckpt doesn't exist, can't save.

你可能感兴趣的:(tensorflow-CNN保存模型)