fix rename error when checkpoint name has '_'

yoshida-lab · May 24, 2021 · f8d7a51 · f8d7a51
1 parent b3fb46f
commit f8d7a51
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xenonpy/model/training/extension/persist.py b/xenonpy/model/training/extension/persist.py
@@ -110,7 +110,7 @@ def on_checkpoint(self, checkpoint: Trainer.checkpoint_tuple, trainer: Trainer)
         if self.only_best_states:
             tmp = checkpoint.id.split('_')
             if tmp[-1] == '1':
-                key = tmp[0]
+                key = '_'.join(tmp)
                 value = deepcopy(checkpoint._asdict())
                 self._checker.set_checkpoint(**{key: value})
         else: