File tree 3 files changed +7
-2
lines changed 3 files changed +7
-2
lines changed Original file line number Diff line number Diff line change @@ -114,6 +114,9 @@ jobs:
114
114
115
115
- script : |
116
116
set -e
117
+ python setup.py install --user
118
+ rm -rf pytorch_lightning
119
+ pip list
117
120
python -m pytest pl_examples -v --maxfail=2 --durations=0
118
121
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
119
122
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
Original file line number Diff line number Diff line change 15
15
fail-fast : false
16
16
# max-parallel: 6
17
17
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] #
19
20
python-version : [3.6, 3.9]
20
21
21
22
steps :
Original file line number Diff line number Diff line change 14
14
_PACKAGE_ROOT = os .path .dirname (_EXAMPLES_ROOT )
15
15
_DATASETS_PATH = os .path .join (_PACKAGE_ROOT , 'Datasets' )
16
16
17
+ _TORCHVISION_MNIST_AVAILABLE = False
17
18
_DALI_AVAILABLE = _module_available ("nvidia.dali" )
18
19
19
20
try :
20
21
from torchvision .datasets .mnist import MNIST
21
22
MNIST (_DATASETS_PATH , download = True )
22
23
_TORCHVISION_MNIST_AVAILABLE = True
23
24
except HTTPError :
24
- _TORCHVISION_MNIST_AVAILABLE = False
25
+ pass
25
26
26
27
LIGHTNING_LOGO = """
27
28
####
You can’t perform that action at this time.
0 commit comments