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