Pytorch进行训练时,使用Tensorboardx的SummaryWriter:
from tensorboardX import SummaryWriter
#定义一个writer用于记录各种数据
writer = SummaryWriter(log_dir=my_path)#my_path为保存数据路径
for i in range(epochs):
writer.add_scalar(train/loss, train_loss, i) #train_loss 为损失函数标量
writer.close()
在Terminal下启动tensorboard:
tensorboard --logdir=/dqwldata1/cdk/seismicdeeplearning/seismic_facies_division/runs --host=167.64.110.14 --port=6006
或者
tensorboard --logdir=my_path
my_path的保存路径如下: