tensorboard可视化

最新推荐文章于 2023-08-05 21:53:06 发布

原创

最新推荐文章于 2023-08-05 21:53:06 发布 · 897 阅读

4 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #机器学习 #python

因为在做实验的时候,需要记录数据,为了方便记录数据,可以使用tensorboard在运行的过程中就将实验结果记录其中,以对比那次实验的结果是最好的.
tensorboard常用的功能有,保存模型网络结构,保存训练集的损失,下面就介绍tensorboard的几个常用的功能.

1.tensorboard面板介绍

这个面板图源自:https://blog.csdn.net/Zhangsan0219/article/details/121763073
我是参考这个作者重新写了一遍笔记,便于自己记录.

2.画折现统计图

将多张图分开画

# -*- coding： utf-8 -*-
# @Author ：WenShuai Xiao
# @Time：2022-05-30
# @File：Tensorboard_line.py
import os.path

from torch.utils.tensorboard import SummaryWriter
import numpy as np
from tqdm import tqdm

if __name__=='__main__':
    print('测试开始')
    '''
    画线
    启动命令 tensorboard --logdir=logs
    指定端口 tensorboard --logdir=logs --port=6007
    '''

    # 创建编辑器,保存日志
    # log_dir保存路径
    logs_path=os.path.join("./logs","trian")
    if not os.pa