- adapted the dataset mean and std value to single value not tuple

franck-armand · Oct 11, 2024 · d24beac · d24beac
1 parent 19dd90c
commit d24beac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/params.py b/params.py
@@ -4,8 +4,8 @@
 data_root = "data"
 dataset_mean_value = 0.5
 dataset_std_value = 0.5
-dataset_mean = (dataset_mean_value, dataset_mean_value, dataset_mean_value)
-dataset_std = (dataset_std_value, dataset_std_value, dataset_std_value)
+dataset_mean = (dataset_mean_value,)
+dataset_std = (dataset_std_value,)
 batch_size = 50
 image_size = 64