deepgeek/Matcha-TTS
2
0
Fork 0
mirror of https://github.com/shivammehta25/Matcha-TTS.git synced 2026-02-05 02:09:21 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adding a bit of comments

Browse Source
This commit is contained in:
Shivam Mehta
2024-02-24 15:20:13 +00:00
parent 8e87111a98
commit 4d5b62cea9
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
matcha/utils/get_durations_from_trained_model.py
View File

@@ -42,7 +42,7 @@ def compute_durations(data_loader: torch.utils.data.DataLoader, model: nn.Module
device (torch.device): GPU or CPU device (torch.device): GPU or CPU
""" """
for batch in tqdm(data_loader, leave=False): for batch in tqdm(data_loader, desc="🍵 Computing durations 🍵:"):
x, x_lengths = batch["x"], batch["x_lengths"] x, x_lengths = batch["x"], batch["x_lengths"]
y, y_lengths = batch["y"], batch["y_lengths"] y, y_lengths = batch["y"], batch["y_lengths"]
spks = batch["spks"] spks = batch["spks"]
@@ -171,4 +171,9 @@ def main():
if __name__ == "__main__": if __name__ == "__main__":
# Helps with generating durations for the dataset to train other architectures
# that cannot learn to align due to limited size of dataset
# Example usage:
# python python matcha/utils/get_durations_from_trained_model.py -i ljspeech.yaml -c pretrained_model
# This will create a folder in data/processed_data/durations/ljspeech with the durations
main() main()