Merge commit 'bfc1cd0756b2e012f1529dfc368345f0a33faad0' into HEAD

ouioui199 · Jan 10, 2025 · 0dfa3c5 · 0dfa3c5
2 parents 2de2752 + bfc1cd0
commit 0dfa3c5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/torchcvnn/datasets/sample/dataset.py b/src/torchcvnn/datasets/sample/dataset.py
@@ -168,6 +168,8 @@ def __getitem__(self, index: int) -> Tuple[torch.Tensor, int]:
             ]
         }
 
+        meta = {k: v[0][0].astype(np.float32) for k, v in meta.items()}
+
         complex_img = data["complex_img"][:, :, np.newaxis]
 
         class_idx = self.class_names.index(key)