From 18d5e0f8f5828cc2e327ef90d712dd805d573222 Mon Sep 17 00:00:00 2001
From: chun <zhengchh6@mail2.sysu.edu.cn>
Date: Fri, 22 Dec 2023 00:41:34 +0800
Subject: [PATCH] workers modified

---
 best_number_workers.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

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