返回介绍

绘制训练曲线

发布于 2024-06-23 17:58:49 字数 2030 浏览 0 评论 0 收藏 0

PARL集成了tensorboardX作为可视化的工具。工具导入方法:

from parl.utils import summary

折线图接口:add_scalar

summary.add_scalar(tag, scalar_value, global_step=None)

常用的参数

  • tag (string) – Data identifier

  • scalar_value (float or string/blobname) – Value to save

  • global_step (int) – Global step value to record

例子:

from parl.utils import summary

x = range(100)
for i in x:
    summary.add_scalar('y=2x', i * 2, i)

预期结果:

https://www.wenjiangs.com/wp-content/uploads/2024/docimg5/add_scalar.jpg

柱形接口:add_histogram

summary.add_histogram(tag, values, global_step=None)

常用的参数

  • tag (string) – Data identifier

  • values (torch.Tensor, numpy.array, or string/blobname) – Values to build histogram

  • global_step (int) – Global step value to record

例子:

from parl.utils import summary
import numpy as np

for i in range(10):
    x = np.random.random(1000)
    summary.add_histogram('distribution centers', x + i, i)

预期结果:

https://www.wenjiangs.com/wp-content/uploads/2024/docimg5/add_histogram.jpg

修改数据保存路径接口

默认的数据保存路径是: ./train_log , summary的数据路径和logger绑定到一起的,所以直接修改logger的保存路径即可:

from parl.utils import logger
logger.set_dir('./train_log/exp1')

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
    我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
    原文