Fix label info on loading checkpoint (#4200)

* Update on_load_checkpoint `LabelInfo`

Author: eugene123tw

CHANGELOG.md

@@ -66,6 +66,10 @@ All notable changes to this project will be documented in this file.
   (<https://github.com/openvinotoolkit/training_extensions/pull/4173>)
 - Fix arrow support for semantic segmentation task
   (<https://github.com/openvinotoolkit/training_extensions/pull/4172>)
+- Fix backward compatibility issues in model checkpoint loading
+  (<https://github.com/openvinotoolkit/training_extensions/pull/4199>)
+- Fix label info on loading checkpoint
+  (<https://github.com/openvinotoolkit/training_extensions/pull/4200>)
 
 ## \[2.2.2\]
 

src/otx/core/model/base.py

@@ -394,7 +394,7 @@ def on_load_checkpoint(self, checkpoint: dict[str, Any]) -> None:
                 ckpt_label_info = LabelInfo(
                     label_groups=ckpt_label_info.label_groups,
                     label_names=ckpt_label_info.label_names,
-                    label_ids=[str(i) for i in range(len(ckpt_label_info.label_names))],
+                    label_ids=ckpt_label_info.label_names,
                 )
             self._label_info = ckpt_label_info