We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 09dc8df commit 5f24766Copy full SHA for 5f24766
wildlife_tools/features/deep.py
@@ -120,9 +120,6 @@ def __call__(self, dataset: ImageDataset) -> FeatureDataset:
120
self.model = self.model.to(self.device)
121
self.model = self.model.eval()
122
123
- # TODO: this is hacky
124
- dataset.transforms = None # Reset transforms.
125
-
126
check_dataset_output(dataset, check_label=False)
127
loader = torch.utils.data.DataLoader(
128
dataset,
0 commit comments