add_scalar和add_scalars
这两个函数是用来保存程序中的数据,然后利用tensorboard工具来进行可视化的。
add_scalar
sw.add_scalar('lr' , optimizer.param_groups[0]['lr'], global_step=global_step)
这行代码的含义就是利用add_scalar函数来生成一个“lr”名称的二维函数,横轴X值为“optimizer.param_groups[0][‘lr’]”,纵轴Y值为global_step。
即: