shixian.shi преди 2 години
родител
ревизия
8d169f9cd9
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      funasr/torch_utils/load_pretrained_model.py

+ 1 - 1
funasr/torch_utils/load_pretrained_model.py

@@ -120,6 +120,6 @@ def load_pretrained_model(
     if ignore_init_mismatch:
         src_state = filter_state_dict(dst_state, src_state)
 
-    logging.info("Loaded src_state keys: {}".format(src_state.keys()))
+    # logging.info("Loaded src_state keys: {}".format(src_state.keys()))
     dst_state.update(src_state)
     obj.load_state_dict(dst_state)