diff options
Diffstat (limited to 'src/text_recognizer/networks/wide_resnet.py')
| -rw-r--r-- | src/text_recognizer/networks/wide_resnet.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/src/text_recognizer/networks/wide_resnet.py b/src/text_recognizer/networks/wide_resnet.py index d1c8f9a..618f414 100644 --- a/src/text_recognizer/networks/wide_resnet.py +++ b/src/text_recognizer/networks/wide_resnet.py @@ -28,10 +28,10 @@ def conv_init(module: Type[nn.Module]) -> None: classname = module.__class__.__name__ if classname.find("Conv") != -1: nn.init.xavier_uniform_(module.weight, gain=np.sqrt(2)) - nn.init.constant(module.bias, 0) + nn.init.constant_(module.bias, 0) elif classname.find("BatchNorm") != -1: - nn.init.constant(module.weight, 1) - nn.init.constant(module.bias, 0) + nn.init.constant_(module.weight, 1) + nn.init.constant_(module.bias, 0) class WideBlock(nn.Module): @@ -183,7 +183,7 @@ class WideResidualNetwork(nn.Module): else None ) - self.apply(conv_init) + # self.apply(conv_init) def _configure_wide_layer( self, in_planes: int, out_planes: int, stride: int, activation: str |