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、结果