我正在使用Tensorflow对象检测API,通过传递学习进行培训。当运行Tensorboard以可视化度量时,为每个显示的两个唯一点分别位于步骤0和最后一步(因此是一条线)。如何获取每个时期或迭代的信息?
训练在进行中,因此我假设在调用model_main.py或经过预训练的模型配置文件(在我的情况下为ssd_mobilenet_v2_coco_2018_03_29.config)中,它必须是一个参数。
python -m object_detection.model_main \
--pipeline_config_path={PIPELINE_CONFIG_PATH} \
--model_dir={MODEL_DIR} \
--num_train_steps=1000 \
--alsologtostderr)
在最初为空的MODEL_DIR文件夹中,我得到了以下信息: