Merge pull request #215 from TommyZihao/patch-2

Update biggan-deep_128x128_cvt_hugging-face_rgb.py
open-mmlab · Jan 11, 2022 · 401cb5a · 401cb5a
2 parents a52cd38 + 823242c
commit 401cb5a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/configs/_base_/models/biggan/biggan-deep_128x128_cvt_hugging-face_rgb.py b/configs/_base_/models/biggan/biggan-deep_128x128_cvt_hugging-face_rgb.py
@@ -9,6 +9,7 @@
         shared_dim=128,
         with_shared_embedding=True,
         sn_eps=1e-6,
+        sn_style='torch',
         init_type='ortho',
         act_cfg=dict(type='ReLU', inplace=True),
         concat_noise=True,
@@ -20,6 +21,7 @@
         num_classes=1000,
         base_channels=128,
         sn_eps=1e-6,
+        sn_style='torch',
         init_type='ortho',
         act_cfg=dict(type='ReLU', inplace=True),
         with_spectral_norm=True),