From 44cced0ebeb79bf889d7c4ab6feda409fef19ec6 Mon Sep 17 00:00:00 2001 From: Ezpgkx9o7 Date: Sat, 9 Oct 2021 15:01:04 +0800 Subject: [PATCH] Update model_define.py --- model_define.py | 1 + 1 file changed, 1 insertion(+) diff --git a/model_define.py b/model_define.py index 3268fde..2975d8d 100644 --- a/model_define.py +++ b/model_define.py @@ -101,6 +101,7 @@ class ResNet(nn.Module): # Zero-initialize the last BN in each residual branch, # so that the residual branch starts with zeros, and each residual block behaves like an identity. # This improves the model by 0.2~0.3% according to https://arxiv.org/abs/1706.02677 + #加入归一化层,对是否进行归一化进行判断 if zero_init_residual: for m in self.modules(): if isinstance(m, Bottleneck):