perf(train): specify num_workers in DataLoader (#313)

src/so_vits_svc_fork/__main__.py

@@ -2,6 +2,7 @@
 
 import os
 from logging import getLogger
+from multiprocessing import freeze_support
 from pathlib import Path
 from typing import Literal
 
@@ -819,4 +820,6 @@ def train_cluster(
     )
 
 
-cli()
+if __name__ == "__main__":
+    freeze_support()
+    cli()

src/so_vits_svc_fork/train.py

@@ -2,6 +2,7 @@
 
 import warnings
 from logging import getLogger
+from multiprocessing import cpu_count
 from pathlib import Path
 from typing import Any
 
@@ -41,16 +42,18 @@ def __init__(self, hparams: Any):
         self.val_dataset = TextAudioDataset(self.__hparams, is_validation=True)
 
     def train_dataloader(self):
-        # since dataset just reads data from a file, set num_workers to 0
         return DataLoader(
             self.train_dataset,
+            # pin_memory=False,
+            num_workers=min(cpu_count(), 4),
             batch_size=self.__hparams.train.batch_size,
             collate_fn=self.collate_fn,
         )
 
     def val_dataloader(self):
         return DataLoader(
             self.val_dataset,
+            # pin_memory=False,
             batch_size=1,
             collate_fn=self.collate_fn,
         )