From 2983009e57d21f0f30cda482e5548212ef4b0f94 Mon Sep 17 00:00:00 2001
From: Shen-Chenhui <e0052680@u.nus.edu>
Date: Fri, 19 Apr 2024 15:34:37 +0800
Subject: [PATCH] config

---
 configs/vae_magvit_v2/train/16x128x128.py       | 4 ++--
 configs/vae_magvit_v2/train/16x128x128_trial.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/configs/vae_magvit_v2/train/16x128x128.py b/configs/vae_magvit_v2/train/16x128x128.py
index 5be8214..9e71b8d 100644
--- a/configs/vae_magvit_v2/train/16x128x128.py
+++ b/configs/vae_magvit_v2/train/16x128x128.py
@@ -74,9 +74,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 = 10
+epochs = 200
 log_every = 1
-ckpt_every = 1000
+ckpt_every = 200
 load = None
 
 batch_size = 4
diff --git a/configs/vae_magvit_v2/train/16x128x128_trial.py b/configs/vae_magvit_v2/train/16x128x128_trial.py
index 7cd5d80..767e6d5 100644
--- a/configs/vae_magvit_v2/train/16x128x128_trial.py
+++ b/configs/vae_magvit_v2/train/16x128x128_trial.py
@@ -73,7 +73,7 @@ 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 = 100
+epochs = 200
 log_every = 1
 ckpt_every = 200
 load = None