pytorch使用tensorboard

tensorboard本质上抽取的是numpy的数据，如果要跟tensor直接做match的话，必须把tensor转换成numpy数据
 

 

tensorboard.py

from tensorboardX import SummaryWriter

writer = SummaryWriter("logs")

for i in range(100):
    writer.add_scalar('y=2x',2*i,i)

writer.close()

运行后就会生成一个logs文件夹

然后在命令行下输入

tensorboard --logdir=logs

#想换端口的话
tensorboard --logdir=logs --port=6007

 

 

示例