使用 TensorBoard 应用程序加载和可视化输出张量 - 亚马逊 SageMaker AI
在 TensorBoard 中显示输出张量

本文属于机器翻译版本。若本译文内容与英语原文存在差异，则一律以英文原文为准。

使用 TensorBoard 应用程序加载和可视化输出张量

通过将从 S3 存储桶收集的输出张量与训练期间或之后的训练作业成对加载，可以进行在线或离线分析。

在打开 TensorBoard 应用程序时，TensorBoard 将打开并显示 SageMaker AI Data Manager 选项卡。以下屏幕截图显示了 TensorBoard 应用程序中的“SageMaker AI Data Manager”选项卡的完整视图。

注意

首次启动 TensorBoard 应用程序时，可视化插件可能不会出现。在 SageMaker AI Data Manager 插件中选择训练作业后，TensorBoard 应用程序会加载 TensorBoard 数据并填充可视化插件。

注意

在处于不活动状态 1 小时后，TensorBoard 应用程序会自动关闭。如果您想在使用完该应用程序后将其关闭，请务必手动关闭 TensorBoard，以免为托管该应用程序的实例付费。有关如何删除应用程序的说明，请参阅删除未使用的 TensorBoard 应用程序。

“SageMaker AI Data Manager”选项卡视图。

在 SageMaker AI Data Manager 选项卡中，您可以选择任意训练作业，并从 Amazon S3 加载与 TensorBoard 兼容的训练输出数据。

  1. 在搜索训练作业部分，使用筛选条件缩小要查找、加载和可视化的训练作业列表的范围。

  2. 在训练作业列表部分中，使用复选框选择要从中提取数据并可视化以进行调试的训练作业。

  3. 选择添加选定作业。选定作业将显示在跟踪的训练作业部分中，如以下屏幕截图所示。

    跟踪训练作业部分。
注意

SageMaker AI Data Manager 选项卡仅显示使用 TensorBoardOutputConfig 参数配置的训练作业。确保您已使用此参数配置 SageMaker AI 估算器。有关更多信息，请参阅 步骤 2：使用 TensorBoard 输出配置创建 SageMaker 训练估算器对象。

注意

如果您是首次使用 SageMaker AI with TensorBoard，或者在之前的使用中未加载任何数据，则可能不会显示可视化选项卡。在添加训练作业并等待几秒钟后，选择右上角的顺时针圆形箭头来刷新查看器。成功加载作业数据后，可视化选项卡将出现。您也可以使用右上角的刷新按钮旁边的设置按钮来设置为自动刷新。

在 TensorBoard 中显示输出张量

在图形选项卡中，您可以在左侧窗格中查找已加载训练作业的列表。您也可以使用训练作业的复选框来显示或隐藏可视化项。TensorBoard 动态插件是动态激活的，具体取决于您如何将训练脚本设置为包括摘要写入器并传递张量和标量集合的回调，因此，图形选项卡也将动态显示。以下屏幕截图显示了每个选项卡的示例视图，其中包含两个训练作业的可视化，这两个训练作业收集了时间序列、标量、图形、分布和直方图插件的指标。

“时间序列” 选项卡视图

时间序列选项卡视图。

“标量”选项卡视图

标量选项卡视图。

“图表”选项卡视图

图表选项卡视图。

“分配”选项卡视图

分配选项卡视图。

“直方图”选项卡视图

直方图选项卡视图。