From 4f915a51dde9c8f6ee405201841cbea571eebbef Mon Sep 17 00:00:00 2001
From: zhengzangw <zhengzangw@gmail.com>
Date: Thu, 2 May 2024 05:41:39 +0000
Subject: [PATCH] updated config

---
 configs/vae/train/video.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/configs/vae/train/video.py b/configs/vae/train/video.py
index dda530d..02bbb24 100644
--- a/configs/vae/train/video.py
+++ b/configs/vae/train/video.py
@@ -19,7 +19,7 @@ plugin = "zero2"
 # Define model
 model = dict(
     type="VideoAutoencoderPipeline",
-    freeze_vae_2d=True,
+    freeze_vae_2d=False,
     from_pretrained=None,
     vae_2d=dict(
         type="VideoAutoencoderKL",
@@ -39,7 +39,7 @@ kl_loss_weight = 1e-6
 
 mixed_image_ratio = 0.2
 use_real_rec_loss = True
-use_z_rec_loss = True
+use_z_rec_loss = False
 use_image_identity_loss = False
 
 # Others