Feat: colab support

2023-12-30 19:51:05 +09:00
parent 618368f0b9
commit 4108478f54
5 changed files with 305 additions and 5 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -172,7 +172,9 @@ def run():
    collate_fn = TextAudioSpeakerCollate()
    train_loader = DataLoader(
        train_dataset,
-        num_workers=min(config.train_ms_config.num_workers, os.cpu_count() - 1),
+        # num_workers=min(config.train_ms_config.num_workers, os.cpu_count() - 1),
+        # Slow and often freezes, so use only half of the cores.
+        num_workers=min(config.train_ms_config.num_workers, os.cpu_count() // 2),
        shuffle=False,
        pin_memory=True,
        collate_fn=collate_fn,