模型每次的训练需要耗费很长的时间,每次都从头开始,时间实在是浪费不起。我们可以把TensorFlow训练的模型保存,以便之后需要再还原使用。
设定每五轮保存一次,创建一个文件夹,用于保存模型文件。Python的os库,在这里主要是对文件夹进行操作,检查是否存在和创建。
Saver就是tf提供给我们保存模型的方法,红框中提示用户,每五轮保存一次的文件名,最后再保存一次最新的。
在完整Python代码中的位置。
模型每次的训练需要耗费很长的时间,每次都从头开始,时间实在是浪费不起。我们可以把TensorFlow训练的模型保存,以便之后需要再还原使用。
设定每五轮保存一次,创建一个文件夹,用于保存模型文件。Python的os库,在这里主要是对文件夹进行操作,检查是否存在和创建。
Saver就是tf提供给我们保存模型的方法,红框中提示用户,每五轮保存一次的文件名,最后再保存一次最新的。
在完整Python代码中的位置。