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

Remote STT tokenizer

This commit is contained in:
Eren Gölge 2021-09-30 14:58:26 +00:00
parent 6d3b2d3cdd
commit 55d9209221
1 changed files with 0 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
TTS/trainer.py
View File

@ -19,7 +19,6 @@ from torch import nn
from torch.nn.parallel import DistributedDataParallel as DDP_th from torch.nn.parallel import DistributedDataParallel as DDP_th
from torch.utils.data import DataLoader from torch.utils.data import DataLoader
from TTS.stt.datasets.tokenizer import Tokenizer
from TTS.utils.callbacks import TrainerCallback from TTS.utils.callbacks import TrainerCallback
from TTS.utils.distribute import init_distributed from TTS.utils.distribute import init_distributed
from TTS.utils.generic_utils import ( from TTS.utils.generic_utils import (
@ -103,7 +102,6 @@ class Trainer:
get_data_samples: Callable = None, get_data_samples: Callable = None,
train_samples: List = None, train_samples: List = None,
eval_samples: List = None, eval_samples: List = None,
tokenizer: Tokenizer = None,
cudnn_benchmark: bool = False, cudnn_benchmark: bool = False,
training_assets: Dict = {}, training_assets: Dict = {},
parse_command_line_args: bool = True, parse_command_line_args: bool = True,
@ -237,9 +235,6 @@ class Trainer:
self.use_apex = self._is_apex_available() self.use_apex = self._is_apex_available()
self.use_amp_scaler = self.config.mixed_precision and self.use_cuda self.use_amp_scaler = self.config.mixed_precision and self.use_cuda
# init tokenizer
self.tokenizer = tokenizer
# load data samples # load data samples
if train_samples is None and get_data_samples is None: if train_samples is None and get_data_samples is None:
raise ValueError("[!] `train_samples` and `get_data_samples` cannot both be None.") raise ValueError("[!] `train_samples` and `get_data_samples` cannot both be None.")