From 3607e8a7bb53dff16c555c5d75f07194ef9bfe39 Mon Sep 17 00:00:00 2001 From: alperenunlu Date: Fri, 7 Jun 2024 02:40:50 +0300 Subject: [PATCH 1/2] Remove torchvision_tutorial from validate_tutorials_built.py --- .jenkins/validate_tutorials_built.py | 1 - 1 file changed, 1 deletion(-) 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", From 045f00cae97e12ed1f818e26ee5378c31dfbb94a Mon Sep 17 00:00:00 2001 From: alperenunlu Date: Fri, 7 Jun 2024 03:29:28 +0300 Subject: [PATCH 2/2] Fix multiprocessing in torchvision_tutorial.py --- intermediate_source/torchvision_tutorial.py | 4 ---- 1 file changed, 4 deletions(-) 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 )