linux远程连接下tensorboard的使用
tensorboard --logdir=[yourlogdir]
此处 yourlogdir
应为目录名,目录下为日志文件
//export TMPDIR=某个路径
//在这里我的具体设置如下:
export TMPDIR=~/trial/backup/logs/
mkdir -p $TMPDIR
tensorboard --logdir=[LOG] --host=192.168.9.214 --port=2262
TensorBoard 2.1.1 at http://192.168.9.214:2262/ (Press CTRL+C to quit)
具体代码编写,参考 AAAI-2021-TACT/code/AUC-PR/managers/trainer.py
from torch.utils.tensorboard import SummaryWriter
TensorboardX 教程
从零开始学Pytorch(十八)-- 可视化工具Tensorboard
网友评论