fix: index_root searching

close #1147
2025-12-21 14:09:41 +01:00 · 2023-09-01 14:11:55 +08:00
parent d634c2727e
commit 8ffdcb0128
21 changed files with 59 additions and 76 deletions
--- a/infer/lib/train/utils.py
+++ b/infer/lib/train/utils.py
@@ -34,7 +34,7 @@ def load_checkpoint_d(checkpoint_path, combd, sbd, optimizer=None, load_opt=1):
                new_state_dict[k] = saved_state_dict[k]
                if saved_state_dict[k].shape != state_dict[k].shape:
                    print(
-                        "shape-%s-mismatch|need-%s|get-%s"
+                        "shape-%s-mismatch. need: %s, get: %s"
                        % (k, state_dict[k].shape, saved_state_dict[k].shape)
                    )  #
                    raise KeyError