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 18:02:02 +08:00
parent 45ea2bd29d
commit a485925dbd
1 changed files with 5 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
scripts/inference-debug.py
View file

@ -125,11 +125,11 @@ def main():
# 3.2. move to device & eval
vae = vae.to(device, dtype).eval()
# 4.5. setup optimizer
optimizer = HybridAdam(
filter(lambda p: p.requires_grad, vae.parameters()), lr=cfg.lr, weight_decay=0, adamw_mode=True
)
lr_scheduler = None
# # 4.5. setup optimizer
# optimizer = HybridAdam(
# filter(lambda p: p.requires_grad, vae.parameters()), lr=cfg.lr, weight_decay=0, adamw_mode=True
# )
# lr_scheduler = None
# # 4.6. prepare for training
# if cfg.grad_checkpoint: