7d8b7d679a
* HifiGAN: initial commit Signed-off-by: Felix Kreuk <felixkreuk@gmail.com> * switched to manual optimization, added grads to FilterBanks object, added v1,v2,v3 configurations for the generator added multiple audio/spec examples in wandb Signed-off-by: Felix Kreuk <felixkreuk@gmail.com> * added trg_mel_fn for different mel params in L1 loss Signed-off-by: Felix Kreuk <felixkreuk@gmail.com> * fixed ci checks Signed-off-by: Felix Kreuk <felixkreuk@gmail.com> * moved losses to separate classes, fixed exp_manager in yaml, style fixes Signed-off-by: Felix Kreuk <felixkreuk@gmail.com> * fixed `min_duration` in yamls of MelGAN and HifiGAN Signed-off-by: Felix Kreuk <felixkreuk@gmail.com> * fixed f_max in loss calculation, added docs Signed-off-by: Felix Kreuk <felixkreuk@gmail.com> Co-authored-by: Oleksii Kuchaiev <okuchaiev@users.noreply.github.com> Co-authored-by: Jason <jasoli@nvidia.com>
43 lines
1.7 KiB
Python
43 lines
1.7 KiB
Python
# Copyright (c) 2020, NVIDIA CORPORATION. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from nemo.collections.tts.models.degli import DegliModel
|
|
from nemo.collections.tts.models.ed_mel2spec import EDMel2SpecModel
|
|
from nemo.collections.tts.models.glow_tts import GlowTTSModel
|
|
from nemo.collections.tts.models.hifigan import HifiGanModel
|
|
from nemo.collections.tts.models.melgan import MelGanModel
|
|
from nemo.collections.tts.models.squeezewave import SqueezeWaveModel
|
|
from nemo.collections.tts.models.tacotron2 import Tacotron2Model
|
|
from nemo.collections.tts.models.talknet import TalkNetDursModel, TalkNetSpectModel
|
|
from nemo.collections.tts.models.two_stages import GriffinLimModel, MelPsuedoInverseModel, TwoStagesModel
|
|
from nemo.collections.tts.models.uniglow import UniGlowModel
|
|
from nemo.collections.tts.models.waveglow import WaveGlowModel
|
|
|
|
__all__ = [
|
|
"GlowTTSModel",
|
|
"SqueezeWaveModel",
|
|
"Tacotron2Model",
|
|
"WaveGlowModel",
|
|
"DegliModel",
|
|
"EDMel2SpecModel",
|
|
"TwoStagesModel",
|
|
"GriffinLimModel",
|
|
"MelPsuedoInverseModel",
|
|
"TalkNetDursModel",
|
|
"TalkNetSpectModel",
|
|
"UniGlowModel",
|
|
"MelGanModel",
|
|
"HifiGanModel",
|
|
]
|