Pytorch-Lightning中的日志记录

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