Linux远程连接下Pytorch使用Tensorboardx进行训练过程可视化

it2023-05-29  73

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的保存路径如下:

最新回复(0)