Try to find best setting

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,

train_ms_jp_extra.py

@@ -190,14 +190,15 @@ def run():
     collate_fn = TextAudioSpeakerCollate(use_jp_extra=True)
     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,
         batch_sampler=train_sampler,
+        persistent_workers=True,
         # これもメモリ消費量を減らそうとしてコメントアウト
-        # persistent_workers=True,
         # prefetch_factor=6,
     )  # DataLoader config could be adjusted.
     eval_dataset = None