From 7858579de012502a8e325bcd4ed9f489fa2486ec Mon Sep 17 00:00:00 2001
From: JonnaMat <matthiesenjonna@gmail.com>
Date: Thu, 30 Mar 2023 16:05:59 +0200
Subject: [PATCH] update mean and std of cifar dataset

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

diff --git a/trainer.py b/trainer.py
index 1eef06c4..a87332c9 100644
--- a/trainer.py
+++ b/trainer.py
@@ -85,8 +85,9 @@ def main():
 
     cudnn.benchmark = True
 
-    normalize = transforms.Normalize(mean=[0.485, 0.456, 0.406],
-                                     std=[0.229, 0.224, 0.225])
+    normalize = transforms.Normalize(mean=[0.4914, 0.4822, 0.4465],
+                                     std=[0.247, 0.243, 0.261])
+
 
     train_loader = torch.utils.data.DataLoader(
         datasets.CIFAR10(root='./data', train=True, transform=transforms.Compose([