From dc45231c8a4b80d3898d2fdac11b1bb2d004a1c5 Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Tue, 16 Apr 2024 17:27:41 +0800 Subject: [PATCH] debug --- configs/vae_magvit_v2/train/17x128x128.py | 1 + 1 file changed, 1 insertion(+) diff --git a/configs/vae_magvit_v2/train/17x128x128.py b/configs/vae_magvit_v2/train/17x128x128.py index d357911..cd5480b 100644 --- a/configs/vae_magvit_v2/train/17x128x128.py +++ b/configs/vae_magvit_v2/train/17x128x128.py @@ -45,6 +45,7 @@ discriminator = dict( # loss weights +logvar_init=0.0 kl_loss_weight = 0.000001 perceptual_loss_weight = 0.1 # use vgg is not None and more than 0 discriminator_factor = 1.0 # for discriminator adversarial loss