Pytorch-Lightning中的日志记录
TensorBoardLogger
常用参数
| 参数名称 | 含义 | 默认值 |
|---|---|---|
| save_dir | 保存路径 | |
| name | 该日志记录器的名字 | default |
| log_graph | 向Tensorboard中添加计算图，要求用户模型定义了self.example_input_array属性 | False |
LightningModule.log()用法
常用参数
| 参数名称 | 含义 | 默认值 |
|---|---|---|
| name | 该日志记录的名字 | |
| value | 该日志记录的值 | |
| prog_bar | 是否在进度条上打印日志 | False |
| logger | 将日志写入日志记录器（前提是Trainer已设置日志记录器） | True |
| on_step | 每次网络权重更新都记录一次 | None(False) |
| on_epoch | 每个epoch都记录一次 | None(False) |
| rank_zero_only | 在分布式训练任务中log()函数仅在0号设备有效 | None(False) |
使用注意
- 如果on_step和on_epoch同时使用，那么log()的name会追加_step和__epoch
