Merge pull request #5273 from voxel51/bugfix/sim-sort-patch

Similarity race condition patch
voxel51 · Dec 18, 2024 · ec2c2e8 · ec2c2e8
2 parents 3d5bd81 + 4fdcc9c
commit ec2c2e8
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/fiftyone/utils/torch.py b/fiftyone/utils/torch.py
@@ -573,8 +573,9 @@ def __exit__(self, *args):
             torch.backends.cudnn.benchmark = self._benchmark_orig
             self._benchmark_orig = None
 
-        self._no_grad.__exit__(*args)
-        self._no_grad = None
+        if self._no_grad is not None:
+            self._no_grad.__exit__(*args)
+            self._no_grad = None
 
     @property
     def media_type(self):