fix in mnist0-3

Author: hzavadil98

utils/dataloaders/mnist_0_3.py

@@ -76,8 +76,10 @@ def __init__(
         self.download = download
         self.num_classes = 4
 
-        if self.split == "train" or self.split == "validation":
-            train = True  # used to decide whether to load training or test dataset
+        if self.split == "test":
+            train = False  # used to decide whether to load training or test dataset
+        else:
+            train = True
 
         if not self.download and not self._chech_is_downloaded():
             raise ValueError(