hc/JSCC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

workers modified

Browse Source
This commit is contained in:
chun 2023-12-22 00:41:34 +08:00
parent 9d6480311c
commit 18d5e0f8f5
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
best_number_workers.py
View File

@ -9,11 +9,10 @@ from torch.utils.data import DataLoader, RandomSampler
if __name__ == '__main__': if __name__ == '__main__':
transform = transforms.Compose([ transform = transforms.Compose([
torchvision.transforms.ToTensor(), torchvision.transforms.ToTensor(),
torchvision.transforms.Normalize((0.1307,), (0.3081,))
]) ])
trainset = torchvision.datasets.MNIST( trainset = torchvision.datasets.CIFAR10(
root='./Dataset/MNIST/', root='./Dataset/',
train=True, # 如果为True，从 training.pt 创建数据，否则从 test.pt 创建数据。 train=True, # 如果为True，从 training.pt 创建数据，否则从 test.pt 创建数据。
download=True, # 如果为true，则从 Internet 下载数据集并将其放在根目录中。 如果已下载数据集，则不会再次下载。 download=True, # 如果为true，则从 Internet 下载数据集并将其放在根目录中。 如果已下载数据集，则不会再次下载。
transform=transform transform=transform