From ccc98d46c9e982c27d05b96c27882d22a5edda0c Mon Sep 17 00:00:00 2001
From: NM512 <morihira3513@gmail.com>
Date: Sat, 17 Jun 2023 15:32:27 +0900
Subject: [PATCH] applied weight_init for linear layer in Decoder

---
 networks.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/networks.py b/networks.py
index aade584..1ba7673 100644
--- 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 = []