Disable multi-processing in all tests and enable torch 1.0 in travis

inferno-pytorch · Dec 20, 2018 · 3106760 · 3106760
1 parent 8e3eaff
commit 3106760
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -7,8 +7,7 @@ python:
   - 3.7
 
 env:
-  # FIXME pytorch 1.0 multi-processing is broken ...
-  # - PYTORCH_CONDA="pytorch" TORCHVISION_CONDA="torchvision" TORCHVISION_CHANNEL=pytorch
+  - PYTORCH_CONDA="pytorch" TORCHVISION_CONDA="torchvision" TORCHVISION_CHANNEL=pytorch
   - PYTORCH_CONDA="pytorch=0.4.1" TORCHVISION_CONDA="torchvision" TORCHVISION_CHANNEL=pytorch
 
 install:

diff --git a/tests/test_training/test_callbacks/test_logging/test_tensorboard.py b/tests/test_training/test_callbacks/test_logging/test_tensorboard.py
@@ -51,9 +51,9 @@ def get_random_dataloaders(self, input_channels=3):
 
         # Build dataloaders from dataset
         train_loader = DataLoader(train_dataset, batch_size=1,
-                                  shuffle=True, num_workers=1, pin_memory=False)
+                                  shuffle=True, num_workers=0, pin_memory=False)
         test_loader = DataLoader(test_dataset, batch_size=1,
-                                 shuffle=True, num_workers=1, pin_memory=False)
+                                 shuffle=True, num_workers=0, pin_memory=False)
         return train_loader, test_loader
 
     def get_trainer(self, input_channels):