def __init__(self, ngpu, **kwargs):
super(netD_images, self).__init__()
self.ngpu = ngpu
self.L = kwargs['L']
self.K = kwargs['K']
self.arguments = kwargs['arguments']
self.l1 = nn.Linear(self.L, self.K, bias=True)
initializationhelper(self.l1, 'tanh')
self.l1_bn = nn.BatchNorm1d(self.K)
self.l2 = nn.Linear(self.K, self.K, bias=True)
initializationhelper(self.l2, 'relu')
#self.l2_bn = nn.BatchNorm1d(self.K)
self.l3 = nn.Linear(self.K, 1, bias=True)
initializationhelper(self.l3, 'relu')
评论列表
文章目录