Skip to content

Commit 67819bb

Browse files
alperenunlusvekars
andauthored
Enable torchvision_tutorial (#2909)
* Remove torchvision_tutorial from validate_tutorials_built.py * Fix multiprocessing in torchvision_tutorial.py --------- Co-authored-by: Svetlana Karslioglu <svekars@meta.com>
1 parent 48905cf commit 67819bb

File tree

2 files changed

+0
-5
lines changed

2 files changed

+0
-5
lines changed

.jenkins/validate_tutorials_built.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,6 @@
2626
"beginner_source/text_sentiment_ngrams_tutorial", # not building with 2.3 RC, might be able to turn on with GA
2727
"beginner_source/t5_tutorial", # re-enable after this is fixed: https://github.com/pytorch/text/issues/1756
2828
"intermediate_source/mnist_train_nas", # used by ax_multiobjective_nas_tutorial.py
29-
"intermediate_source/torchvision_tutorial", # disable due to RuntimeError: DataLoader worker (pid(s) 20092) exited unexpectedly
3029
"intermediate_source/fx_conv_bn_fuser",
3130
"intermediate_source/_torch_export_nightly_tutorial", # does not work on release
3231
"advanced_source/super_resolution_with_onnxruntime",

intermediate_source/torchvision_tutorial.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -382,14 +382,12 @@ def get_transform(train):
382382
# expects during training and inference time on sample data.
383383
import utils
384384

385-
386385
model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights="DEFAULT")
387386
dataset = PennFudanDataset('data/PennFudanPed', get_transform(train=True))
388387
data_loader = torch.utils.data.DataLoader(
389388
dataset,
390389
batch_size=2,
391390
shuffle=True,
392-
num_workers=4,
393391
collate_fn=utils.collate_fn
394392
)
395393

@@ -433,15 +431,13 @@ def get_transform(train):
433431
dataset,
434432
batch_size=2,
435433
shuffle=True,
436-
num_workers=4,
437434
collate_fn=utils.collate_fn
438435
)
439436

440437
data_loader_test = torch.utils.data.DataLoader(
441438
dataset_test,
442439
batch_size=1,
443440
shuffle=False,
444-
num_workers=4,
445441
collate_fn=utils.collate_fn
446442
)
447443

0 commit comments

Comments
 (0)