Tensorflow 对象的最大迭代 API with resnet faster r-cnn

Max iteration of Tensorflow object API with resnet faster r-cnn

我正在使用 ResNet101 Faster R-CNN 教程训练 Oxford 数据集。

我 运行 在我的本地机器上使用 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(我在 ResNet 中使用 COCO 预训练数据)外,我没有更改任何参数。

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

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

注意:根据 Jonathan 的评论,您还可以明确添加步数 num_steps