From d94abdb8773dffbf51f22112b17eb433b8c8b6ec Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Fri, 19 Apr 2024 15:16:14 +0800 Subject: [PATCH] save old config during trainin --- configs/vae_magvit_v2/inference/16x128x128.py | 3 ++- configs/vae_magvit_v2/inference/pipeline_16x128x128.py | 3 ++- configs/vae_magvit_v2/train/16x128x128_trial.py | 1 + configs/vae_magvit_v2/train/pipeline_16x128x128.py | 1 + 4 files changed, 6 insertions(+), 2 deletions(-) diff --git a/configs/vae_magvit_v2/inference/16x128x128.py b/configs/vae_magvit_v2/inference/16x128x128.py index 28b8f01..6d20a91 100644 --- a/configs/vae_magvit_v2/inference/16x128x128.py +++ b/configs/vae_magvit_v2/inference/16x128x128.py @@ -33,7 +33,8 @@ model = dict( kl_embed_dim = 64, activation_fn = 'swish', separate_first_frame_encoding = False, - custom_conv_padding = None + custom_conv_padding = None, + encoder_double_z = False, ) diff --git a/configs/vae_magvit_v2/inference/pipeline_16x128x128.py b/configs/vae_magvit_v2/inference/pipeline_16x128x128.py index 24ad8f6..540d63d 100644 --- a/configs/vae_magvit_v2/inference/pipeline_16x128x128.py +++ b/configs/vae_magvit_v2/inference/pipeline_16x128x128.py @@ -42,7 +42,8 @@ model = dict( activation_fn = 'swish', separate_first_frame_encoding = False, disable_space = True, - custom_conv_padding = None + custom_conv_padding = None, + encoder_double_z = False, ) diff --git a/configs/vae_magvit_v2/train/16x128x128_trial.py b/configs/vae_magvit_v2/train/16x128x128_trial.py index 9de6e7f..7cd5d80 100644 --- a/configs/vae_magvit_v2/train/16x128x128_trial.py +++ b/configs/vae_magvit_v2/train/16x128x128_trial.py @@ -31,6 +31,7 @@ model = dict( activation_fn = 'swish', separate_first_frame_encoding = False, disable_space = False, + encoder_double_z = False, custom_conv_padding = None ) diff --git a/configs/vae_magvit_v2/train/pipeline_16x128x128.py b/configs/vae_magvit_v2/train/pipeline_16x128x128.py index cab30fc..6da9ee1 100644 --- a/configs/vae_magvit_v2/train/pipeline_16x128x128.py +++ b/configs/vae_magvit_v2/train/pipeline_16x128x128.py @@ -36,6 +36,7 @@ model = dict( activation_fn = 'swish', separate_first_frame_encoding = False, disable_space = True, + ncoder_double_z = False, custom_conv_padding = None )