data shapes retrieval and hparam passing

datasets/binar_masks.py

@@ -15,6 +15,10 @@ class BinaryMasksDataset(Dataset):
     _to_label['mask'] = V.MASK
     settings = ['test', 'devel', 'train']
 
+    @property
+    def sample_shape(self):
+        return self[0][0].shape
+
     def __init__(self, data_root, setting, transforms=None):
         assert isinstance(setting, str), f'Setting has to be a string, but was: {self.settings}.'
         assert setting in self.settings, f'Setting must match one of: {self.settings}.'