From 73cec14e45cdbc1945e2191074150715abfde038 Mon Sep 17 00:00:00 2001 From: Runingtime <958919277@qq.com> Date: Wed, 31 May 2023 15:41:14 +0800 Subject: [PATCH] Fix IndexError when loading checkpoints --- rl4lms/envs/text_generation/warm_start.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rl4lms/envs/text_generation/warm_start.py b/rl4lms/envs/text_generation/warm_start.py index 5b4d8326..404a5514 100644 --- a/rl4lms/envs/text_generation/warm_start.py +++ b/rl4lms/envs/text_generation/warm_start.py @@ -97,7 +97,8 @@ def _get_recent_ckpt_path(self, tracker: Tracker): except: os.makedirs(tracker.checkpoint_base_path) checkpoints = os.listdir(tracker.checkpoint_base_path) - + + checkpoints = [cp for cp in checkpoints if "_" in cp] if len(checkpoints) == 0: return None, None