在使用yunyang tensorfow-yolo3 訓(xùn)練數(shù)據(jù)集時，發(fā)現(xiàn)權(quán)重文件最多只能保存十個，訓(xùn)練到后面時，發(fā)現(xiàn)先前生成的權(quán)重文件都沒有了。。。

后來查閱tf.train.Saver()類時，發(fā)現(xiàn)有這樣的描述：

所以，程序就是使用max_to_keep參數(shù)來控制生成權(quán)重文件的最大保存數(shù)量的，如果想保存更多的權(quán)重文件，修改它的值就OK了，在train.py文件中，我這里將max_to_keep修改成100：

with tf.name_scope('loader_and_saver'):self.loader = tf.train.Saver(self.net_var)# D·C 191120：max_to_keep用于控制保存生成權(quán)重文件的個數(shù)，難怪我之前訓(xùn)練的時候前面的權(quán)重文件都沒了！# D·R 191120：將保存的權(quán)重文件個數(shù)改成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()類

總結(jié)

以上是生活随笔為你收集整理的yunyang tensorfow-yolo3 训练时权重文件消失的原因和解决办法（max_to_keep）的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。