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 index 97a099154..4804fded7 100644 --- 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),