Selaa lähdekoodia

Disable multiprocess for Windows. Closes #15.

Yichao Zhou 6 vuotta sitten
vanhempi
commit
eccf0e49d6
2 muutettua tiedostoa jossa 3 lisäystä ja 2 poistoa
  1. 2 1
      process.py
  2. 1 1
      train.py

+ 2 - 1
process.py

@@ -25,6 +25,7 @@ import os.path as osp
 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)

+ 1 - 1
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(