mirror of
https://github.com/hanzoai/mugen.git
synced 2026-11-03 01:48:40 +00:00
29 lines
687 B
Python
29 lines
687 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
"""Loss related classes and functions. In particular the loss balancer from
|
|
EnCodec, and the usual spectral losses."""
|
|
|
|
# flake8: noqa
|
|
from .balancer import Balancer
|
|
from .sisnr import SISNR
|
|
from .stftloss import (
|
|
LogSTFTMagnitudeLoss,
|
|
MRSTFTLoss,
|
|
SpectralConvergenceLoss,
|
|
STFTLoss
|
|
)
|
|
from .specloss import (
|
|
MelSpectrogramL1Loss,
|
|
MultiScaleMelSpectrogramLoss,
|
|
)
|
|
|
|
from .wmloss import (
|
|
WMDetectionLoss,
|
|
WMMbLoss
|
|
)
|
|
|
|
from .loudnessloss import TFLoudnessRatio
|