如何使用tensorboardX保存以及可视化训练过程数据

保存数据： 在代码中： 首先实例化一个SummaryWriter import tensorboardX as tb writer = tb.SummaryWriter(log_dir=tb_path) 然后将需要保存的数据以键-值的方式保存在writer中： writer.add_scalar(key, value, step) 其中step表示了保存的值来自第几次迭代，比如我第三次迭代的tra

>>阅读原文<<