天才一秒记住【长江书屋】地址：https://www.cjshuwu.com
从前有座山，叫巴里赫德，他一个当一座山，十分的爽。
哎，想不到吧。
这个山里面有树林，有庙，有这个山里灵活的狗。
山上有棵树，这棵树不叫高树，因为这个梗太老了。
这棵树的形状有些奇特，大概就长这个样子。
importtorch
fromtorchimportnn
importtorch.nn.functionalasF
importos
importtensorboardX
fromtorch.utils.dataimportDataset
fromtorch.utils.dataimportDataLoader
device=torch.device（'cuda'iftorchuda.is_available（）else'cpu'）
classVGGBaseSimpleS2（nn.Module）:
def__init__（self）:
super（VGGBaseSimpleS2，self）.__init__（）
selfonv1=nn.Sequential（
nn.Conv2d（1，12，kernel_size=3，stride=1，padding=1），
#nn.BatchNorm2d（16），
nn.ReLU（）
）
#6*6
self.max_pooling1=nn.MaxPool2d（kernel_size=2，stride=1）
#5*5
selfonv2_1=nn.Sequential（
nn.Conv2d（12，24，kernel_size=3，stride=1，padding=1），
nn.ReLU（）
）
self.max_pooling2_1=nn.MaxPool2d（kernel_size=2，stride=1）
#4*4
selfonv2_2=nn.Sequential（
nn.Conv2d（24，24，kernel_size=3，stride=1，padding=1），
nn.ReLU（）
）
self.max_pooling2=nn.MaxPool2d（kernel_size=2，stride=2）
#2*2
#2*2
本章未完，请点击下一章继续阅读！若浏览器显示没有新章节了，请尝试点击右上角↗️或右下角↘️的菜单，退出阅读模式即可，谢谢！