Fix multiprocessing in torchvision_tutorial.py

Author: alperenunlu

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,15 +431,13 @@ def get_transform(train):
     dataset,
     batch_size=2,
     shuffle=True,
-    num_workers=4,
     collate_fn=utils.collate_fn
 )
 
 data_loader_test = torch.utils.data.DataLoader(
     dataset_test,
     batch_size=1,
     shuffle=False,
-    num_workers=4,
     collate_fn=utils.collate_fn
 )