从前有座山,叫巴里赫德,他一个当一座山,十分的爽。
哎,想不到吧。
这个山里面有树林,有庙,有这个山里灵活的狗。山上有棵树,这棵树不叫高树,因为这个梗太老了。这棵树的形状有些奇特,大概就长这个样子。
importtorch
fromtorchimportnn
importos
importtensorboardx
.is_availableelsecpu
classvggbasesimples2nn.module:
def__init__self:
supervggbasesimples2,self.__init__
nn.conv2d1,12,kernel_size3,stride1,padding1,
#nn.batchnorm2d16,
nn.relu
#6*6
,stride1
#5*5
nn.conv2d12,24,kernel_size3,stride1,padding1,
nn.relu
,stride1
#4*4
nn.conv2d24,24,kernel_size3,stride1,padding1,
nn.relu
,stride2
#2*2
#2*2
*2*2,2
defforwardself,x:
batchsizex.size0
outself.conv1x
outself.max_pooling1out
outself.conv2_1out
outself.conv2_2out
outself.max_pooling2out
outout.viewbatchsize,-1
outself.fcout
outf.log_softmaxout,dim1