在卷积后面加个BN归一化特征范围
举个例子
torch.nn.Conv2d(in_channels, x, kernel_size=3, padding=1),
torch.nn.BatchNorm2d(x),
torch.nn.ReLU(inplace=True)
原文的代码的复现是没有BN层,而现在以torch训练不加BN是无法收敛的。
在卷积后面加个BN归一化特征范围
torch.nn.Conv2d(in_channels, x, kernel_size=3, padding=1),
torch.nn.BatchNorm2d(x),
torch.nn.ReLU(inplace=True)
原文的代码的复现是没有BN层,而现在以torch训练不加BN是无法收敛的。