fix bug in optimizer path

2023-04-15 13:27:12 -04:00 · 2023-04-15 13:27:12 -04:00 · d25b9661af
parent e574805326
commit d25b9661af
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -438,8 +438,8 @@ def main(args):
            logging.info(f" Saving optimizer state to {save_path}")
            save_optimizer(optimizer, optimizer_path)

+    optimizer_state_path = None
    try:
-
        # check for a local file
        hf_cache_path = get_hf_ckpt_cache_path(args.resume_ckpt)
        if os.path.exists(hf_cache_path) or os.path.exists(args.resume_ckpt):