tensorflow 模型保存问

来源:互联网 发布:蜂鸟软件科技有限公司 编辑:程序博客网 时间:2024/05/22 09:01

出现 model.ckpt.data-00000-of-00001 文件问题。

解决办法:

from tensorflow.core.protobuf import saver_pb2
saver = tf.train.Saver(write_version=saver_pb2.SaverDef.V1)

1 0
原创粉丝点击