applied weight_init for linear layer in Decoder

2023-06-17 15:32:27 +09:00 · 2023-06-17 15:32:27 +09:00 · ccc98d46c9
commit ccc98d46c9
parent 970d1dc3e9
1 changed files with 1 additions and 0 deletions
--- a/networks.py
+++ b/networks.py
@ -549,6 +549,7 @@ class ConvDecoder(nn.Module):
        self._embed_size = minres**2 * depth * 2 ** (layer_num - 1)

        self._linear_layer = nn.Linear(feat_size, self._embed_size)
+        self._linear_layer.apply(tools.weight_init)
        in_dim = self._embed_size // (minres**2)

        layers = []