updates

2019-12-04 15:15:42 -08:00 · 2019-12-04 15:15:42 -08:00 · 0a04eb9ff1
parent a2dc8a6b5a
commit 0a04eb9ff1
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -199,7 +199,7 @@ def train():

    # Dataloader
    batch_size = min(batch_size, len(dataset))
-    nw = min([os.cpu_count(), batch_size if batch_size > 1 else 0, 16])
+    nw = min([os.cpu_count(), batch_size if batch_size > 1 else 0, 16])  # number of workers
    print('Using %g dataloader workers' % nw)
    dataloader = torch.utils.data.DataLoader(dataset,
                                             batch_size=batch_size,