From fb64216b45f2f5a4199a7ccef3dfd1c53af328d5 Mon Sep 17 00:00:00 2001 From: Shen-Chenhui Date: Mon, 8 Apr 2024 17:32:30 +0800 Subject: [PATCH] debug --- scripts/train-vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/train-vae.py b/scripts/train-vae.py index 5e850a7..2f7e5a1 100644 --- a/scripts/train-vae.py +++ b/scripts/train-vae.py @@ -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"))