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 15:34:37 +08:00
parent d94abdb877
commit 2983009e57
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
configs/vae_magvit_v2/train/16x128x128.py
View file

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

2
configs/vae_magvit_v2/train/16x128x128_trial.py
View file

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