笔者使用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
即可使用