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