Try to find best setting

2024-02-07 13:42:20 +09:00
parent d249ba54a0
commit 78689f6c2a
2 changed files with 5 additions and 3 deletions
--- a/train_ms.py
+++ b/train_ms.py
@@ -187,8 +187,9 @@ def run():
    collate_fn = TextAudioSpeakerCollate()
    train_loader = DataLoader(
        train_dataset,
-        # メモリ消費量を減らそうとnum_workersを0にしてみる
+        # メモリ消費量を減らそうとnum_workersを1にしてみる
        # num_workers=min(config.train_ms_config.num_workers, os.cpu_count() // 2),
+        num_workers=1,
        shuffle=False,
        pin_memory=True,
        collate_fn=collate_fn,