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 @@ -121,4 +121,6 @@ jobs:
121121 # cd pl_examples/basic_examples
122122 # bash submit_ddp_job.sh
123123 # bash submit_ddp2_job.sh
124+ env:
125+ PL_USE_MOCKED_MNIST: "1"
124126 displayName: 'Examples'
Original file line number Diff line number Diff line change 1515_DATASETS_PATH = os .path .join (_PACKAGE_ROOT , 'Datasets' )
1616
1717_TORCHVISION_AVAILABLE = _module_available ("torchvision" )
18- _TORCHVISION_MNIST_AVAILABLE = True
18+ _TORCHVISION_MNIST_AVAILABLE = not bool ( os . environ . get ( "PL_USE_MOCKED_MNIST" , False ))
1919_DALI_AVAILABLE = _module_available ("nvidia.dali" )
2020
21- if _TORCHVISION_AVAILABLE :
21+ if _TORCHVISION_MNIST_AVAILABLE :
2222 try :
2323 from torchvision .datasets .mnist import MNIST
2424 MNIST (_DATASETS_PATH , download = True )
You can’t perform that action at this time.
0 commit comments