如何使用tensorboardX保存以及可視化訓練過程數據

保存數據： 在代碼中： 首先實例化一個SummaryWriter import tensorboardX as tb writer = tb.SummaryWriter(log_dir=tb_path) 然後將需要保存的數據以鍵-值的方式保存在writer中： writer.add_scalar(key, value, step) 其中step表示了保存的值來自第幾次迭代，比如我第三次迭代的tra

>>阅读原文<<