def param(nnet, Mb=True):
    """
    Return number parameters(not bytes) in nnet
    """
    neles = sum([param.nelement() for param in nnet.parameters()])
    return neles / 10**6 if Mb else neles

    Return number parameters(not bytes) in nnet
    
train/target_speaker_extraction/models/SpEx_plus/SpEx_plus.py