File tree Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Expand file tree Collapse file tree 2 files changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -103,4 +103,6 @@ jobs:
103
103
# cd pl_examples/basic_examples
104
104
# bash submit_ddp_job.sh
105
105
# bash submit_ddp2_job.sh
106
+ env:
107
+ PL_USE_MOCKED_MNIST: "1"
106
108
displayName: 'Examples'
Original file line number Diff line number Diff line change 15
15
_DATASETS_PATH = os .path .join (_PACKAGE_ROOT , 'Datasets' )
16
16
17
17
_TORCHVISION_AVAILABLE = _module_available ("torchvision" )
18
- _TORCHVISION_MNIST_AVAILABLE = True
18
+ _TORCHVISION_MNIST_AVAILABLE = not bool ( os . environ . get ( "PL_USE_MOCKED_MNIST" , False ))
19
19
_DALI_AVAILABLE = _module_available ("nvidia.dali" )
20
20
21
- if _TORCHVISION_AVAILABLE :
21
+ if _TORCHVISION_MNIST_AVAILABLE :
22
22
try :
23
23
from torchvision .datasets .mnist import MNIST
24
24
MNIST (_DATASETS_PATH , download = True )
You can’t perform that action at this time.
0 commit comments