def __init__(self):
ksize = 3
bn = True
act = F.relu
self.dr = 0.3
super(Z2CNN, self).__init__(
l1=ConvBNAct(
conv=F.Convolution2D(in_channels=1, out_channels=20, ksize=ksize, stride=1, pad=0),
bn=bn,
act=act
),
l2=ConvBNAct(
conv=F.Convolution2D(in_channels=20, out_channels=20, ksize=ksize, stride=1, pad=0),
bn=bn,
act=act
),
l3=ConvBNAct(
conv=F.Convolution2D(in_channels=20, out_channels=20, ksize=ksize, stride=1, pad=0),
bn=bn,
act=act
),
l4=ConvBNAct(
conv=F.Convolution2D(in_channels=20, out_channels=20, ksize=ksize, stride=1, pad=0),
bn=bn,
act=act
),
l5=ConvBNAct(
conv=F.Convolution2D(in_channels=20, out_channels=20, ksize=ksize, stride=1, pad=0),
bn=bn,
act=act
),
l6=ConvBNAct(
conv=F.Convolution2D(in_channels=20, out_channels=20, ksize=ksize, stride=1, pad=0),
bn=bn,
act=act
),
top=F.Convolution2D(in_channels=20, out_channels=10, ksize=4, stride=1, pad=0),
)
评论列表
文章目录