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

pipeline config

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

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

@ -63,7 +63,7 @@ lecam_loss_weight = None # NOTE: MAVGIT v2 use 0.001
# generator_loss_type="non-saturating"
discriminator_loss_type="hinge"
generator_loss_type="hinge"
discriminator_start = 30000 # 50000 NOTE: change to correct val, debug use -1 for now
discriminator_start = 1000 # 50000 NOTE: change to correct val, debug use -1 for now
gradient_penalty_loss_weight = None # 10 # SCH: MAGVIT uses 10, opensora plan doesn't use
ema_decay = 0.999 # ema decay factor for generator
@ -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 = 10
epochs = 500
log_every = 1
ckpt_every = 1000
ckpt_every = 100
load = None
batch_size = 4