Use the actual CIFAR10 normalization values

Ricardo Decal · Ricardo Decal · commit 64bc12eb09f3 · 2025-12-09T16:50:30.000-08:00
diff --git a/beginner_source/hyperparameter_tuning_tutorial.py b/beginner_source/hyperparameter_tuning_tutorial.py
@@ -69,7 +69,7 @@
 
 def load_data(data_dir="./data"):
     transform = transforms.Compose(
-        [transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]
+        [transforms.ToTensor(), transforms.Normalize((0.4914, 0.48216, 0.44653), (0.2022, 0.19932, 0.20086))]
     )
 
     trainset = torchvision.datasets.CIFAR10(

Original file line number	Diff line number	Diff line change
`@@ -69,7 +69,7 @@`
`69`	`69`
`70`	`70`	`def load_data(data_dir="./data"):`
`71`	`71`	`transform = transforms.Compose(`
`72`		`- [transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))]`
	`72`	`+ [transforms.ToTensor(), transforms.Normalize((0.4914, 0.48216, 0.44653), (0.2022, 0.19932, 0.20086))]`
`73`	`73`	`)`
`74`	`74`
`75`	`75`	`trainset = torchvision.datasets.CIFAR10(`