pytorch快速入门（十）cifar-10模型结构可视化

最新推荐文章于 2024-08-25 10:47:03 发布

半甜田田

最新推荐文章于 2024-08-25 10:47:03 发布

阅读量2.3k

点赞数 3

分类专栏：深度学习可视化 pytorch 文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/rensweet/article/details/122364325

版权

本文介绍了如何使用PyTorch构建CIFAR-10模型，并通过代码解释模型结构。通过Tensorboard进行可视化，详细说明了设置padding和stride的计算过程，以及如何运行和查看结果。

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

前言

框架环境：pycharm；使用pytorch环境，请自行配置
可视化工具：tensorboard
有不懂的可翻阅pytorch快速入门之前的笔记

pycharm相关快捷键
ctrl+/ 注释
Ctrl+P 查看参数，如下，光标移动至想要查看的函数，然后按下快捷键

CIFAR-10 Model Structure

下图是cifar-10模型结构
在这里插入图片描述

代码

import torch
from torch import nn
from torch.nn import Conv2d, MaxPool2d, Flatten, Linear, Sequential
from torch.utils.tensorboard import SummaryWriter

#构建模型
class Classification_network(nn.Module):
    def __init__(self):
       super(Classification_network, self).__init__()
       #注1>>>>第一个卷积层,计算出stride=1，padding=2
       # self.conv1 = Conv2