作者:辽河儿女 | 来源:互联网 | 2022-12-02 17:08
我目前正在使用Tensorflow的Saver
类保存和恢复神经网络模型,如下所示:
saver.save(sess, checkpoint_prefix, global_step=step)
saver.restore(sess, checkpoint_file)
这会将.ckpt
模型的文件保存到指定的路径.因为我正在进行多次实验,所以我保留这些模型的空间有限.
我想知道是否有一种方法可以保存这些模型而不保存指定目录中的内容.
防爆.我可以在最后一个检查点将一些对象传递给某个evaluate()函数并从该对象恢复模型吗?
据我所知,save_path
参数in tf.train.Saver.restore()
不是可选的.
任何见解都会非常感激.
谢谢