代码之家 › 专栏 › 技术社区 › yann

“tf.估计器.训练和评估”功能控制训练和评估周期的机制是什么?

object-detection-api tensorflow-estimator tensorflow python

0

yann · 技术社区 · 7 年前

TensorFlow Object Detection API 在最初的10分钟里,它训练速度很快,大约完成了500步(即0.83步/秒)。然后,它减慢速度,大约需要40~50分钟来执行单个训练步骤,在评估数据集上评估模型,并将检查点保存在磁盘上。所以我在几步之后就中断了训练,继续恢复训练。
每次训练前10分钟,它都训练得很快,然后如图所示急剧减速。

模型的训练由 TensorFlow's Estimator API tf.estimator.train_and_evaluate()

1 回复 | 直到 7 年前

1

kvish 7 年前

如果你看 EvalSpec 和 TrainSpec this heated discussion InMemoryEvaluationHook 讨论中提到的这一点可以帮助您,因为它不会在每次调用求值时重新加载检查点。