雾聪
2024-03-13 acf8b4d3dc590121d1297a82ccbb8ad87e9935d9
funasr/train_utils/load_pretrained_model.py
@@ -7,7 +7,7 @@
import torch
import torch.nn
import torch.optim
import pdb
def filter_state_dict(
   dst_state: Dict[str, Union[float, torch.Tensor]],
@@ -63,6 +63,7 @@
   dst_state = obj.state_dict()
   
   print(f"ckpt: {path}")
   if oss_bucket is None:
      src_state = torch.load(path, map_location=map_location)
   else: