tensorboard使用方法
教程链接:https://blog.51cto.com/u_15521344/5056937
注意:
- 一个SummaryWriter只会在文件夹中生成一个events文件(.add_scalars实际上是多个SummaryWriter)
- 一个SummaryWriter可以给不同的tag画图,但是画出的图都不在同一个图里
- 想要画在同一个图中,必须指定相同的tag,指定不同的global_step(一幅图)或者SummaryWriter(多幅图)
常用方式
log_writer.add_scalars(main_tag='train-val', tag_scalar_dict={'train_loss':train_metrics[0],'train_acc':train_metrics[1],'val_acc':val_acc},
global_step=epoch+1)
for name, param in net.named_parameters():
param_writer.add_histogram(tag=name+'_grad', values=param.grad, global_step=epoch)
param_writer.add_histogram(tag=name+'_data', values=param.data, global_step=epoch)
本文来自博客园,作者:SXQ-BLOG,转载请注明原文链接:https://www.cnblogs.com/sxq-blog/p/16640880.html