fix: 卸载音色省显存

顺便将所有print换成了统一的logger

infer/lib/train/utils.py

@@ -33,7 +33,7 @@ def load_checkpoint_d(checkpoint_path, combd, sbd, optimizer=None, load_opt=1):
             try:
                 new_state_dict[k] = saved_state_dict[k]
                 if saved_state_dict[k].shape != state_dict[k].shape:
-                    print(
+                    logger.warn(
                         "shape-%s-mismatch. need: %s, get: %s"
                         % (k, state_dict[k].shape, saved_state_dict[k].shape)
                     )  #
@@ -109,7 +109,7 @@ def load_checkpoint(checkpoint_path, model, optimizer=None, load_opt=1):
         try:
             new_state_dict[k] = saved_state_dict[k]
             if saved_state_dict[k].shape != state_dict[k].shape:
-                print(
+                logger.warn(
                     "shape-%s-mismatch|need-%s|get-%s"
                     % (k, state_dict[k].shape, saved_state_dict[k].shape)
                 )  #
@@ -207,7 +207,7 @@ def latest_checkpoint_path(dir_path, regex="G_*.pth"):
     f_list = glob.glob(os.path.join(dir_path, regex))
     f_list.sort(key=lambda f: int("".join(filter(str.isdigit, f))))
     x = f_list[-1]
-    print(x)
+    logger.debug(x)
     return x