Tensorboard（4）训练分类算法的tensorboard可视化示例

洪流之源

已于 2022-10-18 08:54:09 修改

阅读量696

点赞数

分类专栏： pytorch 文章标签：分类深度学习计算机视觉

于 2022-09-23 19:24:50 首次发布

本文链接：https://blog.csdn.net/weicao1990/article/details/127016722

版权

本文通过一个实例展示了如何使用Tensorboard进行深度学习模型的训练过程可视化，包括训练数据图像、模型计算图、学习率、损失、准确率、梯度和权重参数的展示。

摘要由CSDN通过智能技术生成

该示例实现了tensorboard对训练数据图像、模型计算图、学习率、损失值、准确率、梯度、模型权值参数的可视化：

# 监控loss, accuracy, weights, gradients
import numpy as np
import math
import os
import torch
import torch.nn as nn
import torch.optim as optim
import torchvision
from torchvision import datasets
from torchvision import models
import torchvision.transforms as transforms
from torch.utils.tensorboard import SummaryWriter
from tqdm import tqdm

if __name__ == '__main__':

    # 参数设置
    NUM_CLASSES = 100
    MAX_EPOCH = 100
    BATCH_SIZE = 64
    INIT_LR = 0.01
    LRF = 0.1 # 倍率因子，最后学习率会降到初始学习率的0.1倍
    LOG_INTERVAL = 10
    VAL_INTERVAL = 1
    SAVE_CHECKPOINTS_DIR = 'checkpoints'

    # 构建 SummaryWriter
    summary_writer = SummaryWriter(log_dir='logs')

    if not os.path.exists(SAVE_CHECKPOINTS_DIR):
        os.makedirs(SAVE_CHECKPOINTS_DIR)
    
    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

    transform_train = transforms.Compose([transforms.Resize(256),
                                         transforms.CenterCrop(224), 
                                         transforms.ToTensor(),
                                         transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225])])
 
    transform_test = transforms.Compose([transforms.Resize(256),
                                         transforms.CenterCrop(224), 
                                         transforms.ToTensor(),
                                         transforms.Normalize(mean=[0.485, 0.456, 0.406], s