Update pages/1_TensorIntro.py

pages/1_TensorIntro.py

@@ -213,7 +213,7 @@ import torchvision.transforms as transforms
 # Define the transformation for the dataset
 transform = transforms.Compose(
     [transforms.ToTensor(),
-     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])
+     transforms.Normalize((0.5, 0.5, 0.5))])
 
 # Load the CIFAR-10 dataset
 trainset = torchvision.datasets.CIFAR10(root='./data', train=True, download=True, transform=transform)
@@ -282,7 +282,6 @@ with torch.no_grad():
         correct += (predicted == labels).sum().item()
 
 print(f'Accuracy of the network on the 10000 test images: {100 * correct / total}%')
-
     
 '''
     },