nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-21 11:59:01 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

debug

Browse source
This commit is contained in:
Shen-Chenhui 2024-04-08 17:19:48 +08:00
parent a634f7327e
commit d6c0131aab
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
configs/vae_3d/inference/16x256x256.py
View file

@ -43,7 +43,14 @@ model = dict(
# Others
seed = 42
save_dir = "outputs/samples"
batch_size = 8
wandb = False
epochs = 5
log_every = 1
ckpt_every = 1000
load = None
batch_size = 8
grad_clip = 1.0
grad_checkpoint = True
lr = 1e-4