数据并行可能导致的模型训练报错 (#2139)

pointerhacker

2024-10-15 70645e48072bf193fbf069949f1d2b10fddac8a3

 funasr/frontends/utils/beamformer.py

@@ -76,9 +76,7 @@
    return beamform_vector


def apply_beamforming_vector(
    beamform_vector: ComplexTensor, mix: ComplexTensor
) -> ComplexTensor:
def apply_beamforming_vector(beamform_vector: ComplexTensor, mix: ComplexTensor) -> ComplexTensor:
    # (..., C) x (..., C, T) -> (..., T)
    es = FC.einsum("...c,...ct->...t", [beamform_vector.conj(), mix])
    return es