From c8ec57e255376c63cd8241a4a2de423af16a1dcc Mon Sep 17 00:00:00 2001 From: Phillip Chlap Date: Thu, 7 Dec 2023 09:28:43 +1100 Subject: [PATCH] Correct typo --- platipy/imaging/cnn/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/platipy/imaging/cnn/dataset.py b/platipy/imaging/cnn/dataset.py index fb06d086..69c48136 100644 --- a/platipy/imaging/cnn/dataset.py +++ b/platipy/imaging/cnn/dataset.py @@ -630,7 +630,7 @@ def __getitem__(self, index): if context_map is not None: context_map = torch.FloatTensor(context_map) - context_map = context_map.unqsqueeze(0) + context_map = context_map.unsqueeze(0) label = torch.FloatTensor( np.concatenate([np.expand_dims(l, 0) for l in labels], 0)