失眠网,内容丰富有趣,生活中的好帮手!
失眠网 > yunyang tensorfow-yolo3 训练时权重文件消失的原因和解决办法(max_to_keep)

yunyang tensorfow-yolo3 训练时权重文件消失的原因和解决办法(max_to_keep)

时间:2021-06-02 18:26:35

相关推荐

yunyang tensorfow-yolo3 训练时权重文件消失的原因和解决办法(max_to_keep)

在使用yunyang tensorfow-yolo3训练数据集时,发现权重文件最多只能保存十个,训练到后面时,发现先前生成的权重文件都没有了。。。

后来查阅tf.train.Saver()类时,发现有这样的描述:

所以,程序就是使用max_to_keep参数来控制生成权重文件的最大保存数量的,如果想保存更多的权重文件,修改它的值就OK了,在train.py文件中,我这里将max_to_keep修改成100:

with tf.name_scope('loader_and_saver'):self.loader = tf.train.Saver(_var)# D·C 191120:max_to_keep用于控制保存生成权重文件的个数,难怪我之前训练的时候前面的权重文件都没了!# D·R 191120:将保存的权重文件个数改成100:# origin:self.saver = tf.train.Saver(tf.global_variables(), max_to_keep=10)self.saver = tf.train.Saver(tf.global_variables(), max_to_keep=100)

参考文章:tensorflow tf.train.Saver()类

如果觉得《yunyang tensorfow-yolo3 训练时权重文件消失的原因和解决办法(max_to_keep)》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。