nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-14 18:25:35 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

config

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-19 09:28:11 +08:00
parent df6c068496
commit fde35c77c0
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
configs/vae_magvit_v2/train/pipeline_16x128x128.py
View file

@ -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