From a9c5ca1aace845bb0e92197ee3985b0b42ca9e77 Mon Sep 17 00:00:00 2001 From: Tommy in Tongji <36354458+TommyZihao@users.noreply.github.com> Date: Tue, 11 Jan 2022 16:46:07 +0800 Subject: [PATCH] Update biggan-deep_128x128_cvt_hugging-face_rgb.py --- .../models/biggan/biggan-deep_128x128_cvt_hugging-face_rgb.py | 2 ++ 1 file changed, 2 insertions(+) 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),