Estimator的model_fn包含params参数,但params不会传递给Estimator

Question

我正在尝试在本地运行Object Detection API.

我相信我已按照TensorFlow对象检测API文档中的描述设置了所有内容,但是,当我尝试运行model_main.py时,此警告显示并且模型未进行训练.(我无法确定模型是否正在训练,因为该过程没有终止,但没有进一步的日志出现)

警告:tensorflow:Estimator的model_fn(.model_fn位于0x0000024BDBB3D158>)包含params参数,但params不会传递给Estimator.

我传递的代码是:

python tensorflow-models/research/object_detection/model_main.py \
--model_dir=training \
--pipeline_config_path=ssd_mobilenet_v1_coco.config \
--checkpoint_dir=ssd_mobilenet_v1_coco_2017_11_17/model.ckpt \
--num_tain_steps=2000 \
--num_eval_steps=200 \
--alsologtostderr

可能导致此警告的原因是什么？

为什么代码似乎卡住了？

请帮忙!

Answer 1

我遇到了同样的问题,我发现这个警告与模型不起作用的问题无关.我可以使模型工作,因为这个警告显示.

我的错误是我误解了running_locally.md文件中的这一行

"$ {MODEL_DIR}指向将训练检查点和事件写入"的目录"

我将MODEL_DIR更改为{project directory}/models/model目录结构的位置:

model_dir

它奏效了.希望这可以帮助你.