在使用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)》对你有帮助,请点赞、收藏,并留下你的观点哦!