ds
游雁
2024-05-20 62acaf1158e9019cd49089d92175d47989bb1841
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()