先导入包
from torch.utils.tensorboard import SummaryWriter
创建文件log日志
writer = SummaryWriter(log_dir='tensor')
创建所需要看的曲线实例
writer.add_scalar('classification_loss', float(classification_loss), global_step=epoch_num) writer.add_scalar('regression_loss', float(regression_loss), global_step=epoch_num) writer.add_scalar('mean', np.mean(loss_hist), global_step=epoch_num)
在terminal输入
tensorboard --logdir=E:\**\**\tensor 使用绝对路径