在使用Tensorflow對象檢測API進行訓練時，我如何知道步驟總數？

Question

最近3個小時，我一直在GPU驅動的雲機上使用以下命令運行培訓工作：

python legacy/train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_pets.config

然后運行該日志，例如：

 INFO:tensorflow:global step 14455: loss = 0.5896 (0.775 sec/step)
I1001 19:27:43.575182 140054916601600 tf_logging.py:116] global step 14455: loss = 0.5896 (0.775 sec/step)

我如何知道要執行多少個步驟或總共要執行幾個步驟？

Answer 1

如果您使用的是預定義的模型拓撲，則可以在模型隨附的文檔中查找訓練時間段（以時期或步驟為單位）。 如果您創建了自己的模型，則可以通過觀察測試結果來確定培訓時間。 當准確性達到可接受的水平然后開始下降時，您可能會過度訓練。 備份到高精度。 重復此實驗幾次，以確定模型的“最佳位置”。

Answer 2

在ssd_mobilenet_v1_pets.config它在第163行中說：

num_steps: 200000

這是總步驟數，如果您未進行任何更改，培訓腳本將執行。