self.median_stats = {key: (val.median(), torch.median(torch.abs(val - val.median()))) for key, val in self.data.items() 
                             if type(val) is torch.Tensor and val.dim() == 1 and val.is_floating_point()}