从前有座山,叫巴里赫德,他一个当一座山,十分的爽。
哎,想不到吧。
这个山里面有树林,有庙,有这个山里灵活的狗。山上有棵树,这棵树不叫高树,因为这个梗太老了。这棵树的形状有些奇特,大概就长这个样子。
importtorch
fromtorchimportnn
importtorch.nn.functionalasF
importos
importtensorboardX
fromtorch.utils.dataimportDataset
fromtorch.utils.dataimportDataLoader
device=torch.device('cuda'iftorch.cuda.is_available()else'cpu')
classVGGBaseSimpleS2(nn.Module):
def__init__(self):
super(VGGBaseSimpleS2,self).__init__()
self.conv1=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
self.conv2_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
self.conv2_2=nn.Sequential(
nn.Conv2d