Mobaxterm+pytorch+tensorboard

1.安装tensorboard

pip install tensorboardX
pip install tensorboard
pip install tensorflow

2.显示每层的loss

from tensorboardX import SummaryWriter
#定义Summary_Writer
writer = SummaryWriter('./Result')   # 数据存放在这个文件夹

for i, (name, param) in enumerate(model.named_parameters()):
    if 'bn' not in name:
        writer.add_histogram(name, param, 0)
        writer.add_scalar('loss', loss, i)

3.打开tensonboard

tensorboard --logdir='./Result' --host=127.0.0.1
#参数分别是路径和端口

4.Mobaxterm连接pc与服务器

ifconfig
#得到服务器ip
127.0.0.1用于数据传输
10.XX用于连接

参考https://blog.csdn.net/qq_40662074/article/details/108511862

5.在本地浏览器打开127.0.0.1:16006