diff --git a/.jenkins/validate_tutorials_built.py b/.jenkins/validate_tutorials_built.py index eb027929f48..ebd654b7006 100644 --- a/.jenkins/validate_tutorials_built.py +++ b/.jenkins/validate_tutorials_built.py @@ -26,7 +26,6 @@ "beginner_source/text_sentiment_ngrams_tutorial", # not building with 2.3 RC, might be able to turn on with GA "beginner_source/t5_tutorial", # re-enable after this is fixed: https://github.com/pytorch/text/issues/1756 "intermediate_source/mnist_train_nas", # used by ax_multiobjective_nas_tutorial.py - "intermediate_source/torchvision_tutorial", # disable due to RuntimeError: DataLoader worker (pid(s) 20092) exited unexpectedly "intermediate_source/fx_conv_bn_fuser", "intermediate_source/_torch_export_nightly_tutorial", # does not work on release "advanced_source/super_resolution_with_onnxruntime", diff --git a/intermediate_source/torchvision_tutorial.py b/intermediate_source/torchvision_tutorial.py index f1562d71a47..d1e4c5c5d5e 100644 --- a/intermediate_source/torchvision_tutorial.py +++ b/intermediate_source/torchvision_tutorial.py @@ -382,14 +382,12 @@ def get_transform(train): # expects during training and inference time on sample data. import utils - model = torchvision.models.detection.fasterrcnn_resnet50_fpn(weights="DEFAULT") dataset = PennFudanDataset('data/PennFudanPed', get_transform(train=True)) data_loader = torch.utils.data.DataLoader( dataset, batch_size=2, shuffle=True, - num_workers=4, collate_fn=utils.collate_fn ) @@ -433,7 +431,6 @@ def get_transform(train): dataset, batch_size=2, shuffle=True, - num_workers=4, collate_fn=utils.collate_fn ) @@ -441,7 +438,6 @@ def get_transform(train): dataset_test, batch_size=1, shuffle=False, - num_workers=4, collate_fn=utils.collate_fn )