fix loss normalization for ddp training

haoneng.lhn

2023-12-08 acb9a0fec8d8a4dabeedcbb8e08c26f66d7083f0

 funasr/layers/stft.py

@@ -4,8 +4,11 @@
from typing import Union

import torch
from torch_complex.tensor import ComplexTensor

try:
    from torch_complex.tensor import ComplexTensor
except:
    print("Please install torch_complex firstly")
from funasr.modules.nets_utils import make_pad_mask
from funasr.layers.complex_utils import is_complex
from funasr.layers.inversible_interface import InversibleInterface