add grayscale for fashionmnist

Author: Demoren1

code/DARTS_baseline.py

@@ -3,7 +3,6 @@
 import os
 import logging
 from pathlib import Path
-import shutil
 
 import argparse
 import torch

code/train_models.py

@@ -76,7 +76,7 @@ def __init__(self, config: TrainConfig):
             config.device if torch.cuda.is_available() else "cpu"
         )
 
-    def get_data_loaders(self, batch_size: int = None, seed=None) -> Tuple[DataLoader, DataLoader]:
+    def get_data_loaders(self, batch_size: int = None, seed=None) -> Tuple[DataLoader, DataLoader, DataLoader]:
         """
         Create training and validation data loaders for the chosen dataset.
         """
@@ -93,6 +93,7 @@ def get_data_loaders(self, batch_size: int = None, seed=None) -> Tuple[DataLoade
         test_transform = transforms.Compose(
             self.base_transform
             + [
+                transforms.Grayscale(num_output_channels=3),
                 transforms.ToTensor(),
                 transforms.Normalize(self.MEAN, self.STD),
             ]