Disable multiprocess for Windows. Closes #15.

zhou13 · Jan 10, 2020 · eccf0e4 · eccf0e4
1 parent 0082f17
commit eccf0e4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/process.py b/process.py
@@ -25,6 +25,7 @@
 import threading
 import subprocess
 
+import yaml
 import numpy as np
 import torch
 import matplotlib as mpl
@@ -85,7 +86,7 @@ def main():
         shuffle=False,
         batch_size=M.batch_size,
         collate_fn=collate,
-        num_workers=C.io.num_workers,
+        num_workers=C.io.num_workers if os.name != "nt" else 0,
         pin_memory=True,
     )
     os.makedirs(args["<output-dir>"], exist_ok=True)

diff --git a/train.py b/train.py
@@ -94,7 +94,7 @@ def main():
     datadir = C.io.datadir
     kwargs = {
         "collate_fn": collate,
-        "num_workers": C.io.num_workers,
+        "num_workers": C.io.num_workers if os.name != "nt" else 0,
         "pin_memory": True,
     }
     train_loader = torch.utils.data.DataLoader(