Tensorboard可视化SSD网络结构

最新推荐文章于 2024-01-28 14:33:06 发布

敲代码的小风

最新推荐文章于 2024-01-28 14:33:06 发布

阅读量594

点赞数

CC 4.0 BY-SA版权

分类专栏：零基础学习SSD网络PyTorch实现 Tensorboard在Pytorch中使用的学习笔记

本文链接：https://blog.csdn.net/m0_46653437/article/details/112382715

零基础学习SSD网络PyTorch实现同时被 2 个专栏收录

293 篇文章

订阅专栏

Tensorboard在Pytorch中使用的学习笔记

15 篇文章

订阅专栏

本文展示了如何使用PyTorch的SSD模型进行训练，并利用TensorBoard实时记录和可视化网络结构。通过逐层输入随机张量，读者可以理解模型内部结构并监控训练过程。

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

from torch.utils.tensorboard import SummaryWriter
from torch.utils.tensorboard import SummaryWriter
import torch
import torchvision
from torchvision import datasets,transforms
from torch.autograd import Variable
from ssd import SSD

ssd = SSD()
model = ssd.net.module
writer = SummaryWriter()
for i in range(5):
    images = torch.randn(4, 3, 300, 300).cuda()
    writer.add_graph(model, input_to_model=images, verbose=False)
writer.flush()
writer.close()