nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-04-24 17:44:47 +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:32:30 +08:00
parent 25707a9d7d
commit fb64216b45
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
scripts/train-vae.py
View file

@ -175,7 +175,7 @@ def main():
if cfg.load is not None:
logger.info("Loading checkpoint")
booster.load_model(vae, os.path.join(cfg.load, "model"))
booster.load_optimizer(optimizer, os.path.join(cfg.load, "optimizer"))
# booster.load_optimizer(optimizer, os.path.join(cfg.load, "optimizer"))
# if lr_scheduler is not None:
# booster.load_lr_scheduler(lr_scheduler, os.path.join(cfg.load, "lr_scheduler"))
running_states = load_json(os.path.join(cfg.load, "running_states.json"))