From 31067607aa9d5b9dcd4be79468cfbbd97a2fe614 Mon Sep 17 00:00:00 2001 From: Constantin Pape Date: Thu, 20 Dec 2018 10:06:59 +0100 Subject: [PATCH] Disable multi-processing in all tests and enable torch 1.0 in travis --- .travis.yml | 3 +-- .../test_callbacks/test_logging/test_tensorboard.py | 4 ++-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/.travis.yml b/.travis.yml index 02ee6be1..4543d36c 100644 --- 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 index 36825890..bb71919e 100755 --- 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):