【pytorch】利用tensorboard记录训练过程与查看
文章目录代码读取保存的log代码from torch.utils.tensorboard import SummaryWriter在epoch内保存,三个参数分别为保存字符串名、保存变量和步数writer.add_scalar('training loss', running_loss / 10, epoch * size + step)读取保存的log命令行输入:tensorboard --logdir= /home/...(events保存地址)...
原创
2021-07-09 11:33:01 ·
3308 阅读 ·
0 评论