用 nn.Tanh() 替换 nn.Sigmoid():深度学习判别器代码优化
class Discriminator(nn.Module): def init(self): super(Discriminator, self).init() self.linear = nn.Sequential( nn.Linear(28*28, 512), nn.LeakyReLU(),
nn.Linear(512, 256),
nn.LeakyReLU(),
nn.Linear(256, 1),
nn.Tanh() # 替换成 nn.Tanh()
)
原文地址: https://www.cveoy.top/t/topic/nF8y 著作权归作者所有。请勿转载和采集!