Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Rename Speaker encoder module to encoder

This commit is contained in:
Edresson Casanova 2022-02-28 15:28:29 -03:00
parent dbe9da7f15
commit 71a1907f4c
9 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
TTS/bin/train_encoder.py
View File

@ -10,11 +10,11 @@ import torch
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
from trainer.torch import NoamLR from trainer.torch import NoamLR
from TTS.speaker_encoder.dataset import SpeakerEncoderDataset from TTS.encoder.dataset import SpeakerEncoderDataset
from TTS.speaker_encoder.losses import AngleProtoLoss, GE2ELoss, SoftmaxAngleProtoLoss from TTS.encoder.losses import AngleProtoLoss, GE2ELoss, SoftmaxAngleProtoLoss
from TTS.speaker_encoder.utils.generic_utils import save_best_model, setup_speaker_encoder_model from TTS.encoder.utils.generic_utils import save_best_model, setup_speaker_encoder_model
from TTS.speaker_encoder.utils.training import init_training from TTS.encoder.utils.training import init_training
from TTS.speaker_encoder.utils.visual import plot_embeddings from TTS.encoder.utils.visual import plot_embeddings
from TTS.tts.datasets import load_tts_samples from TTS.tts.datasets import load_tts_samples
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor
from TTS.utils.generic_utils import count_parameters, remove_experiment_folder, set_init_dict from TTS.utils.generic_utils import count_parameters, remove_experiment_folder, set_init_dict

2
TTS/speaker_encoder/dataset.py
View File

@ -4,7 +4,7 @@ import numpy as np
import torch import torch
from torch.utils.data import Dataset from torch.utils.data import Dataset
from TTS.speaker_encoder.utils.generic_utils import AugmentWAV, Storage from TTS.encoder.utils.generic_utils import AugmentWAV, Storage
class SpeakerEncoderDataset(Dataset): class SpeakerEncoderDataset(Dataset):

2
TTS/speaker_encoder/models/lstm.py
View File

@ -3,7 +3,7 @@ import torch
import torchaudio import torchaudio
from torch import nn from torch import nn
from TTS.speaker_encoder.models.resnet import PreEmphasis from TTS.encoder.models.resnet import PreEmphasis
from TTS.utils.io import load_fsspec from TTS.utils.io import load_fsspec

4
TTS/speaker_encoder/utils/generic_utils.py
View File

@ -8,8 +8,8 @@ from multiprocessing import Manager
import numpy as np import numpy as np
from scipy import signal from scipy import signal
from TTS.speaker_encoder.models.lstm import LSTMSpeakerEncoder from TTS.encoder.models.lstm import LSTMSpeakerEncoder
from TTS.speaker_encoder.models.resnet import ResNetSpeakerEncoder from TTS.encoder.models.resnet import ResNetSpeakerEncoder
from TTS.utils.io import save_fsspec from TTS.utils.io import save_fsspec

2
TTS/tts/utils/speakers.py
View File

@ -9,7 +9,7 @@ import torch
from coqpit import Coqpit from coqpit import Coqpit
from TTS.config import get_from_config_or_model_args_with_default, load_config from TTS.config import get_from_config_or_model_args_with_default, load_config
from TTS.speaker_encoder.utils.generic_utils import setup_speaker_encoder_model from TTS.encoder.utils.generic_utils import setup_speaker_encoder_model
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor

6
tests/aux_tests/test_speaker_encoder.py
View File

@ -3,9 +3,9 @@ import unittest
import torch as T import torch as T
from tests import get_tests_input_path from tests import get_tests_input_path
from TTS.speaker_encoder.losses import AngleProtoLoss, GE2ELoss, SoftmaxAngleProtoLoss from TTS.encoder.losses import AngleProtoLoss, GE2ELoss, SoftmaxAngleProtoLoss
from TTS.speaker_encoder.models.lstm import LSTMSpeakerEncoder from TTS.encoder.models.lstm import LSTMSpeakerEncoder
from TTS.speaker_encoder.models.resnet import ResNetSpeakerEncoder from TTS.encoder.models.resnet import ResNetSpeakerEncoder
file_path = get_tests_input_path() file_path = get_tests_input_path()

2
tests/aux_tests/test_speaker_encoder_train.py
View File

@ -4,7 +4,7 @@ import shutil
from tests import get_device_id, get_tests_output_path, run_cli from tests import get_device_id, get_tests_output_path, run_cli
from TTS.config.shared_configs import BaseAudioConfig from TTS.config.shared_configs import BaseAudioConfig
from TTS.speaker_encoder.speaker_encoder_config import SpeakerEncoderConfig from TTS.encoder.speaker_encoder_config import SpeakerEncoderConfig
def run_test_train(): def run_test_train():

4
tests/aux_tests/test_speaker_manager.py
View File

@ -6,8 +6,8 @@ import torch
from tests import get_tests_input_path from tests import get_tests_input_path
from TTS.config import load_config from TTS.config import load_config
from TTS.speaker_encoder.utils.generic_utils import setup_speaker_encoder_model from TTS.encoder.utils.generic_utils import setup_speaker_encoder_model
from TTS.speaker_encoder.utils.io import save_checkpoint from TTS.encoder.utils.io import save_checkpoint
from TTS.tts.utils.speakers import SpeakerManager from TTS.tts.utils.speakers import SpeakerManager
from TTS.utils.audio import AudioProcessor from TTS.utils.audio import AudioProcessor

2
tests/tts_tests/test_vits.py
View File

@ -7,7 +7,7 @@ from trainer.logging.tensorboard_logger import TensorboardLogger
from tests import assertHasAttr, assertHasNotAttr, get_tests_data_path, get_tests_input_path, get_tests_output_path from tests import assertHasAttr, assertHasNotAttr, get_tests_data_path, get_tests_input_path, get_tests_output_path
from TTS.config import load_config from TTS.config import load_config
from TTS.speaker_encoder.utils.generic_utils import setup_speaker_encoder_model from TTS.encoder.utils.generic_utils import setup_speaker_encoder_model
from TTS.tts.configs.vits_config import VitsConfig from TTS.tts.configs.vits_config import VitsConfig
from TTS.tts.models.vits import Vits, VitsArgs, amp_to_db, db_to_amp, load_audio, spec_to_mel, wav_to_mel, wav_to_spec from TTS.tts.models.vits import Vits, VitsArgs, amp_to_db, db_to_amp, load_audio, spec_to_mel, wav_to_mel, wav_to_spec
from TTS.tts.utils.speakers import SpeakerManager from TTS.tts.utils.speakers import SpeakerManager