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