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)
posted @ 2022-08-30 21:22  SXQ-BLOG  阅读(288)  评论(0编辑  收藏  举报