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 b94c574 commit 02aa93bCopy full SHA for 02aa93b
napari_cellseg3d/code_models/model_framework.py
@@ -337,7 +337,7 @@ def get_device(show=True):
337
def empty_cuda_cache(self):
338
"""Empties the cuda cache if the device is a cuda device"""
339
if self.get_device(show=False).type == "cuda":
340
- logger.info("Attempting to empty cache...")
+ logger.info("Emptying cache...")
341
torch.cuda.empty_cache()
342
logger.info("Attempt complete : Cache emptied")
343
0 commit comments