admin 管理员组

文章数量: 1086019


2024年2月20日发(作者：正则在线登录)

add_scalar用法

`add_scalar`是PyTorch的`SummaryWriter`类的方法之一，用于将标量值添加到TensorBoard的日志中。

`add_scalar`的用法如下：

```python

_scalar(tag, scalar_value, global_step=None,

walltime=None)

```

参数说明：

- `tag`：字符串类型，指定标量值的名称，将在TensorBoard中显示。

- `scalar_value`：数值类型，要记录的标量值。

- `global_step`：可选参数，整数类型，用于指定当前的训练步数或迭代次数。如果不提供，则默认为当前时间戳。

- `walltime`：可选参数，用于指定记录的时间，必须是浮点数，单位为秒。如果不提供，则默认为当前时间。

示例代码：

```python

from board import SummaryWriter

# 创建SummaryWriter对象，指定日志存储在log_dir目录下

writer = SummaryWriter(log_dir='logs')

# 记录训练损失

for epoch in range(100):

train_loss = ...

_scalar('Train Loss', train_loss, epoch)

# 关闭SummaryWriter

()

```

上述代码将使用`add_scalar`方法将训练损失添加到日志中，并使用`epoch`作为全局步数。


本文标签： 指定 训练 用于 记录

版权声明：本文标题：add_scalar用法 内容由网友自发贡献，该文观点仅代表作者本人， 转载请联系作者并注明出处：http://www.roclinux.cn/p/1708403298a522880.html， 本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌抄袭侵权/违法违规的内容，一经查实，本站将立刻删除。