From fde35c77c0c8c801ef7a3ea38d51dada129b4828 Mon Sep 17 00:00:00 2001
From: Shen-Chenhui <e0052680@u.nus.edu>
Date: Fri, 19 Apr 2024 09:28:11 +0800
Subject: [PATCH] config

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

diff --git a/configs/vae_magvit_v2/train/pipeline_16x128x128.py b/configs/vae_magvit_v2/train/pipeline_16x128x128.py
index 90a4280..1e3b157 100644
--- a/configs/vae_magvit_v2/train/pipeline_16x128x128.py
+++ b/configs/vae_magvit_v2/train/pipeline_16x128x128.py
@@ -80,9 +80,9 @@ magvit uses about # samples (K) * epochs ~ 2-5 K,  num_frames = 4, reso = 128
 3-6 epochs for pexel, from pexel observation its correct
 '''
 
-epochs = 500
+epochs = 100
 log_every = 1
-ckpt_every = 100
+ckpt_every = 200
 load = None
 
 batch_size = 4