Skip to content

Commit c5281a3

Browse files
committed
Minor changes
1 parent 8323d11 commit c5281a3

File tree

3 files changed

+7
-2
lines changed

3 files changed

+7
-2
lines changed

.azure-pipelines/gpu-tests.yml

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,9 @@ jobs:
114114
115115
- script: |
116116
set -e
117+
python setup.py install --user
118+
rm -rf pytorch_lightning
119+
pip list
117120
python -m pytest pl_examples -v --maxfail=2 --durations=0
118121
bash pl_examples/run_examples-args.sh --trainer.gpus 1 --trainer.max_epochs 1 --data.batch_size 64 --trainer.limit_train_batches 5 --trainer.limit_val_batches 3
119122
bash pl_examples/run_ddp-examples.sh --trainer.max_epochs 1 --data.batch_size 32 --trainer.limit_train_batches 2 --trainer.limit_val_batches 2

.github/workflows/ci_pkg-install.yml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@ jobs:
1515
fail-fast: false
1616
# max-parallel: 6
1717
matrix:
18-
os: [ubuntu-20.04, macOS-10.15, windows-2019]
18+
# PyTorch 1.5 is failing on Win and bolts requires torchvision>=0.5
19+
os: [ubuntu-20.04, macOS-10.15 , windows-2019] #
1920
python-version: [3.6, 3.9]
2021

2122
steps:

pl_examples/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,14 +14,15 @@
1414
_PACKAGE_ROOT = os.path.dirname(_EXAMPLES_ROOT)
1515
_DATASETS_PATH = os.path.join(_PACKAGE_ROOT, 'Datasets')
1616

17+
_TORCHVISION_MNIST_AVAILABLE = False
1718
_DALI_AVAILABLE = _module_available("nvidia.dali")
1819

1920
try:
2021
from torchvision.datasets.mnist import MNIST
2122
MNIST(_DATASETS_PATH, download=True)
2223
_TORCHVISION_MNIST_AVAILABLE = True
2324
except HTTPError:
24-
_TORCHVISION_MNIST_AVAILABLE = False
25+
pass
2526

2627
LIGHTNING_LOGO = """
2728
####

0 commit comments

Comments
 (0)