3-36Pytorch与tensorboardX

首先建立一个文件夹log

运行一下脚本

from tensorboardX import SummaryWriter

writer = SummaryWriter("log")

for i in range(100):
    writer.add_scalar("a",i,global_step = i)
    writer.add_scalar("b",i ** 2,global_step = i)
writer.close()

在终端输入如下命令

(pytorch_gpu) E:\pytorch>cd log
系统找不到指定的路径。

(pytorch_gpu) E:\pytorch>cd 003

(pytorch_gpu) E:\pytorch\003>cd log

(pytorch_gpu) E:\pytorch\003\log>tensorboard --logdir ./
2021-01-28 23:28:35.357052: I tensorflow/stream_executor/platform/default/dso_loader.cc:44] Successfully
 opened dynamic library cudart64_101.dll
Serving TensorBoard on localhost; to expose to the network, use a proxy or pass --bind_all
TensorBoard 2.2.2 at http://localhost:6006/ (Press CTRL+C to quit)

打开

http://localhost:6006/#scalars

success.