我用的是 tensorflow 1.9 GPU 版本,模型意外中断后想继续训练模型。使用 restore 的时候出现了如题的报错。经过检查,发现是缺乏 checkpoint 文件。

    tensorflow 在保存模型时,会产生以下四个文件:存储网络结构.meta、存储训练好的参数.data和.index、记录最新的模型checkpoint。

    当使用 saver.restore(sess, tf.train.latest_checkpoint(path)) 语句恢复模型时,path 是 checkpoint 所在的路径。如果缺乏 checkpoint 则会报错。

    

Logo

昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链

更多推荐