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

原创已于 2023-03-18 09:34:00 修改 · 548 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #python

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

代码文件专栏收录该内容

10 篇文章 ¥69.90 ¥99.00

订阅专栏

超级会员免费看

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

一、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