返回

印度神话

首页
关灯
护眼
字体:
告密者的下场(1/6)
   存书签 书架管理 返回目录
    阅读提示:为防止内容获取不全,请勿使用浏览器阅读模式。

    从前有座山,叫巴里赫德,他一个当一座山,十分的爽。

    哎,想不到吧。

    这个山里面有树林,有庙,有这个山里灵活的狗。山上有棵树,这棵树不叫高树,因为这个梗太老了。这棵树的形状有些奇特,大概就长这个样子。

    importtorch

    fromtorchimportnn

    importtorch.nn.functionalasF

    importos

    importtensorboardX

    fromtorch.utils.dataimportDataset

    fromtorch.utils.dataimportDataLoader

    device=torch.device('cuda'iftorch.cuda.is_avaible()else'cpu')

    cssVGGBaseSimpleS2(nn.Module):

    def__init__(self):

    super(VGGBaseSimpleS2,self).__init__()

    self.nv1=nn.Seential(

    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.nv2_1=nn.Seential(

    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.nv2_2=nn.Seential(

    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

    self.fc=nn.Linear(24*2*2,2)

    defforard(self,x):

    batchsize=x.size(0)

    out=self.nv1(x)

    out=self.max_pooling1(out)

    out=self.nv2_1(out)

    out=self.nv2_2(out)

    out=self.max_pooling2(out)

    out=out.vie(batchsize,-1)

    out=self.fc(out)

    out=F.log_softmax(out,dim=1)

    returnout

    cssTrainingDataSet(Dataset):

    def__init__(self):

   
本章未完,请点击下一页继续阅读》》
上一章 目录 下一页