From c28c697a0bd068019981b9cfac651522ee08ee73 Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Fri, 19 Apr 2024 09:24:15 +0800 Subject: [PATCH] config for pipeline discriminator --- configs/vae_magvit_v2/train/pipeline_16x128x128.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/configs/vae_magvit_v2/train/pipeline_16x128x128.py b/configs/vae_magvit_v2/train/pipeline_16x128x128.py index a2e7bf4..66859c9 100644 --- a/configs/vae_magvit_v2/train/pipeline_16x128x128.py +++ b/configs/vae_magvit_v2/train/pipeline_16x128x128.py @@ -46,7 +46,8 @@ discriminator = dict( num_frames = num_frames, in_channels = 4, filters = 128, - channel_multipliers = (2,4,4,4,4), # (2,4,4,4) for 64x64 resolution + # channel_multipliers = (2,4,4,4,4), # (2,4,4,4) for 64x64 resolution + channel_multipliers= (2,4,4) # since on intermediate layer, 16 x 16 x 16 dimension z )