diff --git a/tasks/task_3/shipsnet/data.py b/tasks/task_3/shipsnet/data.py index 45966c9..8540abc 100644 --- a/tasks/task_3/shipsnet/data.py +++ b/tasks/task_3/shipsnet/data.py @@ -87,7 +87,7 @@ def setup(self, stage: Union[str, None] = None) -> None: labels = torch.tensor(data_dict["labels"], dtype=torch.bool) # Apply standardising transformations here: - # pixels = ... + pixels = (pixel/255)*2-1 self.full_dataset = LabelledTensorDataset(pixels, labels)