funasr/train_utils/average_nbest_models.py
@@ -62,7 +62,8 @@ # Check if we have any state_dicts to average if len(state_dicts) < 1: raise RuntimeError("No checkpoints found for averaging.") print("No checkpoints found for averaging.") return # Average or sum weights avg_state_dict = OrderedDict()