pytorch的tensorboard使用方法

笔者使用torch版本:1.5.0
载入库的写法为

from torch.utils.tensorboard import SummaryWriter

程序中创建tensorboard的语句为

writer = SummaryWriter(comment=f'LR_{lr}_BS_{batch_size}_SCALE_{img_scale}')

意为自动生成保存路径名称。注释(学习率,batchsize和图像尺寸)加在文件名后面。然后可往writer中添加loss等信息,具体写法网上也容易查到。
但是这样是无法直接运行的,tensorboard的库需要另外安装

pip install tensorboard

即可使用

你可能感兴趣的:(pytorch)