打卡【小土堆】第18/33个视频《卷积层》

本节相机介绍了conv2d的重要参数，其中前五个为最常用的参数

stride和padding的作用可以看动态图：https://github.com/vdumoulin/conv_arithmetic/blob/master/README.md#convolution-animations

我们写一个简单的神经网络，复习之前学过的知识，获取数据集，加载数据集，将数据集写入

import torch
import torchvision
from torch import nn
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter

#准备数据集
dataset = torchvision.datasets.CIFAR10(root='./data', train