update README.md

This commit is contained in:
erogol 2020-06-04 02:58:25 +02:00
parent 6f4c1108f0
commit 8dc1d6a208
2 changed files with 6 additions and 4 deletions

View File

@ -17,9 +17,11 @@ If you are new, you can also find [here](http://www.erogol.com/text-speech-deep-
[Details...](https://github.com/mozilla/TTS/wiki/Mean-Opinion-Score-Results) [Details...](https://github.com/mozilla/TTS/wiki/Mean-Opinion-Score-Results)
## Features ## Features
- High performance Text2Speech models on Torch and Tensorflow 2.0. - High performance Deep Learning models for Text2Speech related tasks.
- High performance Speaker Encoder to compute speaker embeddings efficiently. - Text2Speech models (Tacotron, Tacotron2).
- Integration with various Neural Vocoders (PWGAN, MelGAN, WaveRNN) - Speaker Encoder to compute speaker embeddings efficiently.
- Vocoder models (MelGAN, Multiband-MelGAN, GAN-TTS)
- Ability to convert Torch models to Tensorflow 2.0 for inference.
- Released trained models. - Released trained models.
- Efficient training codes for PyTorch. (soon for Tensorflow 2.0) - Efficient training codes for PyTorch. (soon for Tensorflow 2.0)
- Codes to convert Torch models to Tensorflow 2.0. - Codes to convert Torch models to Tensorflow 2.0.

View File

@ -52,7 +52,7 @@ def setup_loader(ap, is_val=False, verbose=False):
# sampler = DistributedSampler(dataset) if num_gpus > 1 else None # sampler = DistributedSampler(dataset) if num_gpus > 1 else None
loader = DataLoader(dataset, loader = DataLoader(dataset,
batch_size=1 if is_val else c.batch_size, batch_size=1 if is_val else c.batch_size,
shuffle=False, shuffle=True,
drop_last=False, drop_last=False,
sampler=None, sampler=None,
num_workers=c.num_val_loader_workers num_workers=c.num_val_loader_workers