tensorflow - 具有 resnet 更快 r-cnn 的 Tensorflow 对象 API 的最大迭代

Question

我正在使用带有 ResNet101 Faster R-CNN 的教程训练牛津数据集。

我正在使用 1 个 GPU 不使用 Google Cloud 的本地机器上运行训练。

我的问题是我可以知道最大迭代次数是多少吗？

我的脚步已经超过1300万了，还没有停下来。

原始更快的 r-cnn 可以在这里定义最大迭代大小 https://github.com/rbgirshick/py-faster-rcnn/blob/master/tools/train_faster_rcnn_alt_opt.py#L80

，但我不确定 TensorFlow 对象检测 API。

除了 input_path 和 fine_tune_checkpoint（我使用 COCO 预训练数据和 ResNet）之外，我没有更改任何参数。

我认为最大迭代将在配置文件 https://github.com/tensorflow/models/blob/master/object_detection/samples/configs/faster_rcnn_resnet101_pets.config#L100中，但它似乎只在特定步骤后定义学习率.

Answer 1

根据文档 By default, the training job will run indefinitely until the user kills it。因此，同时运行训练和评估作业并终止进程（根据验证准确度饱和提前停止。

注意：从 Jonathan 评论中，您还可以显式添加步数num_steps