Tensorboard的使用（tensorflow和pytorch框架均可用）

Tensorboard的使用

Tensorboard个人使用，害怕忘记了，保存一下.
1.在环境下安装tensorboard

pip install tensorboard
pip install tensorboardX

2.创建logs目录（在项目根目录下）

mkdir ./logs/

3.设置每次实验的日志名称：（这里按照时间来命名）

import datetime
from tensorboardX import SummaryWriter
#from torch.utils.tensorboard import SummaryWriterv  torch也可以这样导入

# 创建一个日志目录
log_dir = "logs/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")  #可以改为config信息
# 初始化SummaryWriter
writer = SummaryWriter(log_dir)

4.在训练过程中，记录变化信息，（loss or acc）:

for epoch in range(num_epochs):
    # 训练模型...
    loss = ...
    accuracy = ...
    
    # 记录损失和准确率
    writer.add_scalar('Loss/train', loss, epoch)
    writer.add_scalar('Accuracy/train', accuracy, epoch)
 
# 关闭writer
writer.close()

4.启动Tensorboard

tensorboard --logdir=logs/ --port=6008  #端口号可省略