pytorch参数可视化tensorboard

1、pip install tensorboard

2、from torch.utils.tensorboard import SummaryWriter

3、 writer = SummaryWriter(comment='test_your_comment', filename_suffix="_test_your_filename_suffix")

4、

for name, param in model.named_parameters():

        writer.add_histogram(tag=name + '_grad', values=param.grad)

        writer.add_histogram(tag=name + '_data', values=param)

这一段代码应该放在train函数里面，说多了都是泪，都怪自己太菜。

5、结果