Python Script запускается сам дважды

Question

Я изучаю PyTorch и прорабатываю учебники.Я обнаружил, что когда я пытаюсь использовать torchvision, мой скрипт запускается дважды.Интересно, если он настроен неправильно, или я случайно ввел какой-то параллелизм.

Вот мой код:

# show images
import torch
import torchvision
import torchvision.transforms as transforms
import matplotlib.pyplot as plt
import numpy as np

transform = transforms.Compose(
    [transforms.ToTensor(),
     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])

trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
                                        download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
                                          shuffle=True, num_workers=2)

testset = torchvision.datasets.CIFAR10(root='./data', train=False,
                                       download=True, transform=transform)
testloader = torch.utils.data.DataLoader(testset, batch_size=4,
                                         shuffle=False, num_workers=2)

classes = ('plane', 'car', 'bird', 'cat',
           'deer', 'dog', 'frog', 'horse', 'ship', 'truck')

def imshow(img):
    img = img / 2 + 0.5     # unnormalize
    npimg = img.numpy()
    plt.imshow(np.transpose(npimg, (1, 2, 0)))
    plt.show()


# get some random training images
print('Type of trainloader')
print(type(trainloader))

dataiter = iter(trainloader)
images, labels = dataiter.next()

# show images
imshow(torchvision.utils.make_grid(images))
# print labels
print(' '.join('%5s' % classes[labels[j]] for j in range(4)))

Мой вывод:

C:\Users\Avi\Anaconda3\python.exe C:/Users/Avi/PycharmProjects/ExampleNN/test.py
Files already downloaded and verified
Files already downloaded and verified
Type of trainloader
<class 'torch.utils.data.dataloader.DataLoader'>
Files already downloaded and verified
Files already downloaded and verified
Type of trainloader
<class 'torch.utils.data.dataloader.DataLoader'>

Изатем происходит сбой при dataiter = iter(trainloader) во второй раз, предположительно потому, что итерация уже завершена.

Кто-нибудь может сказать мне, что я делаю неправильно?

ОБНОВЛЕНИЕ:

Я обнаружил, что если я устанавливаю num_workers = 0 в конструкторе DataLoader, он запускается только один раз и не вызывает сбоев.Если я установлю его на 1, он запускается дважды и вылетает так же, как когда я устанавливаю его на 2. Я все еще не понимаю, почему это происходит, но, по крайней мере, я могу обойти свою проблему.