Tensorflow:打印丢失到日志文件

时间:2018-09-03 12:55:01

标签: python tensorflow

我使用Google CoLab通过以下命令训练模型:

!python /root/models/research/object_detection/model_main.py \
    --pipeline_config_path=/drive/data/ssd_mobilenet_v1_0.75_depth_300x300_coco14_sync.config \
    --model_dir=/drive/data/ \
    --num_train_steps=50000 \
    --num_eval_steps=2000 \
    --alsologtostderr

model_main.py 不显示损失值。使用Tensorboard很好,但是在CoLab上有点困难。

问题

是否至少有一种方法可以将损失值和步骤号打印到stdout(控制台)或日志文件中?

0 个答案:

没有答案