added nr_channels - issue #78

Author: hzavadil98

.gitignore

@@ -9,6 +9,9 @@ wandb/*
 wandb_api.py
 doc/autoapi
 
+*.DS_Store
+
+
 #Magnus specific
 job*
 env2/*

CollaborativeCoding/dataloaders/mnist_0_3.py

@@ -53,6 +53,7 @@ def __init__(
         sample_ids: list,
         train: bool = False,
         transform=None,
+        nr_channels: int = 1,
     ):
         super().__init__()