nettracer3d 0.5.8__py3-none-any.whl → 0.5.9__py3-none-any.whl

nettracer3d/segmenter.py

@@ -1099,8 +1099,11 @@ class InteractiveSegmenter:
     def cleanup(self):
         """Clean up GPU memory"""
         if self.use_gpu:
-            cp.get_default_memory_pool().free_all_blocks()
-            torch.cuda.empty_cache()
+            try:
+                cp.get_default_memory_pool().free_all_blocks()
+                torch.cuda.empty_cache()
+            except:
+                pass
 
     def train_batch(self, foreground_array, speed = True, use_gpu = False, use_two = False):
         """Train directly on foreground and background arrays"""

{nettracer3d-0.5.8.dist-info → nettracer3d-0.5.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nettracer3d
-Version: 0.5.8
+Version: 0.5.9
 Summary: Scripts for intializing and analyzing networks from segmentations of three dimensional images.
 Author-email: Liam McLaughlin <mclaughlinliam99@gmail.com>
 Project-URL: User_Tutorial, https://www.youtube.com/watch?v=cRatn5VTWDY
@@ -44,6 +44,6 @@ NetTracer3D is free to use/fork for academic/nonprofit use so long as citation i
 
 NetTracer3D was developed by Liam McLaughlin while working under Dr. Sanjay Jain at Washington University School of Medicine.
 
--- Version 0.5.8 updates --
+-- Version 0.5.9 updates --
 
-1. Bug fixes and improvements on the 2d segmenter.
+1. Bug fixes

{nettracer3d-0.5.8.dist-info → nettracer3d-0.5.9.dist-info}/RECORD

@@ -10,12 +10,12 @@ nettracer3d/network_draw.py,sha256=F7fw6Pcf4qWOhdKwLmhwqWdschbDlHzwCVolQC9imeU,1
 nettracer3d/node_draw.py,sha256=k3sCTfUCJs3aH1C1q1gTNxDz9EAQbBd1hsUIJajxRx8,9823
 nettracer3d/proximity.py,sha256=FnIiI_AzfXd22HwCIFIyQRZxKYJ8YscIDdPnIv-wsO4,10560
 nettracer3d/run.py,sha256=xYeaAc8FCx8MuzTGyL3NR3mK7WZzffAYAH23bNRZYO4,127
-nettracer3d/segmenter.py,sha256=eHN0JIOznIVY2nV3tlCdLWtTv2-yfQ_PmALw-aTNKAs,51869
+nettracer3d/segmenter.py,sha256=PgKGDTkuQPntTHBuLvSCplOPkGIqd2EQce4l9rhes9I,51938
 nettracer3d/simple_network.py,sha256=fP1gkDdtQcHruEZpUdasKdZeVacoLOxKhR3bY0L1CAQ,15426
 nettracer3d/smart_dilate.py,sha256=Kekm6YIVlJniMvJMG6_AwwNmCqK2l4Qtvg9VzzqPKMw,24600
-nettracer3d-0.5.8.dist-info/LICENSE,sha256=gM207DhJjWrxLuEWXl0Qz5ISbtWDmADfjHp3yC2XISs,888
-nettracer3d-0.5.8.dist-info/METADATA,sha256=wU0DQtE1oYhCJ7D72162tv5wqvzqwLAnU-972TUsk1c,3028
-nettracer3d-0.5.8.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-nettracer3d-0.5.8.dist-info/entry_points.txt,sha256=Nx1rr_0QhJXDBHAQg2vcqCzLMKBzSHfwy3xwGkueVyc,53
-nettracer3d-0.5.8.dist-info/top_level.txt,sha256=zsYy9rZwirfCEOubolhee4TyzqBAL5gSUeFMzhFTX8c,12
-nettracer3d-0.5.8.dist-info/RECORD,,
+nettracer3d-0.5.9.dist-info/LICENSE,sha256=gM207DhJjWrxLuEWXl0Qz5ISbtWDmADfjHp3yC2XISs,888
+nettracer3d-0.5.9.dist-info/METADATA,sha256=NiwydCiq-bMfMv6G1gZajGnrklMW33aW5JQSx11nqu0,2990
+nettracer3d-0.5.9.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+nettracer3d-0.5.9.dist-info/entry_points.txt,sha256=Nx1rr_0QhJXDBHAQg2vcqCzLMKBzSHfwy3xwGkueVyc,53
+nettracer3d-0.5.9.dist-info/top_level.txt,sha256=zsYy9rZwirfCEOubolhee4TyzqBAL5gSUeFMzhFTX8c,12
+nettracer3d-0.5.9.dist-info/RECORD,,