from torch.utils.tensorboard import SummaryWriter
来自官网的示例:TensorBoard: TensorFlow可视化
查看 event 文件
文件名含义
根据参考[1]文档的来说:文件名含义如下:
<WORKING_DIR>/runs/<DATETIME>_<MACHINE_NAME>/events.out.tfevents.<TIMESTAMP>.<MACHINE_NAME>
后面的我想改一下默认值,
每天看这些东西难受,我也分不清楚谁是谁,因为我也不是一次就运行成功的,指能完完整整跑到最后,谁知道哪天中间就断开了,所以会有一堆文件;
在浏览器上查看图像
输入以下命令:
tensorboard --logdir=$EVENTS_FOLDER
$EVENTS_FOLDER
是装有event
文件的文件夹路径,
比如:
output/
|—events.out.tfevents. …
输入:tensorboard --logdir=output
参考
[1] Visualize Model Training with TensorBoard