Update dcgan.py

Embedding layer missing from weight initialization.
DeepTrackAI · May 5, 2024 · 6687191 · 6687191
1 parent ef67856
commit 6687191
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/deeplay/models/discriminators/dcgan.py b/deeplay/models/discriminators/dcgan.py
@@ -20,6 +20,7 @@ def dcgan_discriminator(encoder: ConvolutionalEncoder2d):
         targets=(
             nn.Conv2d,
             nn.BatchNorm2d,
+            nn.Embedding,
             nn.Linear,
         ),
         mean=0,