pytorch——折线图

代码：

from torch.utils.tensorboard import SummaryWriter

def print_hi(name):
    writer=SummaryWriter("logs")
    for i in range(100):
        writer.add_scalar("y=2*x",2*i,i)
        #y=2x为折线图名称、2*i为y轴，i为x轴

    writer.close()

# Press the green button in the gutter to run the script.
if __name__ == '__main__':
    print_hi('PyCharm')

备注：

在terminal处输入   PS E:\2021SOUHU\pythonProject2> tensorboard --logdir=logs

来查看显示端口（6006）

修改端口：PS E:\2021SOUHU\pythonProject2> tensorboard --logdir=logs --port=6007

重新运行时，先删掉log下面的文件，再run，再重新ctrl+c，ensorboard --logdir=logs进入6006端口