Update train.py

#tomcattwo edited synthesizer\train.py per issue CorentinJ#669 and blue-fish/Real-Time-Voice-Cloning@89a9964 to fix Win10 pickle issue

synthesizer/train.py

@@ -15,6 +15,8 @@
 from pathlib import Path
 import sys
 import time
+#tomcattwo added below line per issue #669 and blue-fish/Real-Time-Voice-Cloning@89a9964 to fix Win10 pickle issue
+import platform
 
 
 def np_now(x: torch.Tensor): return x.detach().cpu().numpy()
@@ -142,11 +144,12 @@ def train(run_id: str, syn_dir: str, models_dir: str, save_every: int,
 
         for p in optimizer.param_groups:
             p["lr"] = lr
-
+
+#tomcattwo changed line 152 below per issue #669 and blue-fish/Real-Time-Voice-Cloning@89a9964 to fix Win10 pickle issue
         data_loader = DataLoader(dataset,
                                  collate_fn=lambda batch: collate_synthesizer(batch, r, hparams),
                                  batch_size=batch_size,
-                                 num_workers=2,
+                                 num_workers=2 if platform.system() != "Windows" else 0,
                                  shuffle=True,
                                  pin_memory=True)