0

如何在训练模型时在 Jupyter Notebook 中获取丢失消息？我运行这个命令，没有任何显示（模型训练正常，因为正在硬盘上的“my_model”文件夹中创建新的检查点，并且 GPU 风扇正在努力工作）：

!python model_main_tf2.py --model_dir=models\my_model --pipeline_config_path=models\my_model\pipeline.config

当我直接在 cmd 中运行相同的命令时，每 100 步会出现这样的消息（我想在 Jupyter Notebook 中显示相同的消息）：

INFO:tensorflow:Step 100 per-step time 5.851s
I0120 12:09:11.947002 140647614986112 model_lib_v2.py:707] Step 100 per-step time 5.851s
INFO:tensorflow:{'Loss/classification_loss': 0.24940333,
'Loss/localization_loss': 0.2611718,
'Loss/regularization_loss': 0.15213925,
'Loss/total_loss': 0.66271436,
'learning_rate': 0.0319994}
I0120 12:09:11.947765 140647614986112 model_lib_v2.py:708] {'Loss/classification_loss': 0.24940333,
'Loss/localization_loss': 0.2611718,
'Loss/regularization_loss': 0.15213925,
'Loss/total_loss': 0.66271436,
'learning_rate': 0.0319994}

此外，当我在 Google Colab 中运行相同的命令时，消息会不断显示，并且我会收到培训状态更新。

我在 Windows 10 x64 机器上运行它。

4

0 回答 0