windows下，pytorch ，用tensorboard进行可视化！

土豆娃potato

已于 2023-03-18 09:34:00 修改

阅读量526

点赞数

CC 4.0 BY-SA版权

分类专栏：代码文件文章标签： pytorch 深度学习 python

于 2022-09-14 15:59:38 首次发布

本文链接：https://blog.csdn.net/m0_49234921/article/details/126854151

代码文件专栏收录该内容

10 篇文章 ¥69.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了在Windows下使用PyTorch结合TensorBoard进行深度学习模型的可视化，包括绘制Loss曲线、ACC曲线、参数分布及梯度分布。在过程中遇到了如importlib_metadata导入错误、NotImplementedError、ImportError等问题，文中详细给出了每个问题的解决办法，并提供了Demo演示如何绘制直方图。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、Demo

1、PPT 2、PlotNeuralNet 3、ConvNetDraw

二、windows下，pytorch ，用tensorboard进行可视化！

代码如下：

from tensorboardX import SummaryWriter
import torch
import numpy as np
import math

weights = torch.load(r'weights\yolov3-spp-ultralytics-512.pt')
str