From 721e727287edc9364b8dcf9a8db0cdcb3dc1d8f8 Mon Sep 17 00:00:00 2001 From: schwabjohannes Date: Tue, 5 Mar 2024 12:33:07 +0000 Subject: [PATCH] bug fix --- dynamight/deformations/optimize_deformations.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dynamight/deformations/optimize_deformations.py b/dynamight/deformations/optimize_deformations.py index b8a5f66..a869382 100644 --- a/dynamight/deformations/optimize_deformations.py +++ b/dynamight/deformations/optimize_deformations.py @@ -351,6 +351,7 @@ def optimize_deformations( fits = True print('consensus gaussian models initialized') torch.cuda.empty_cache() + gpu_box = decoder_half1.vol_box except Exception as error: torch.cuda.empty_cache() print( @@ -360,7 +361,7 @@ def optimize_deformations( Ivol = Ivol[0, 0] decoder_half1.vol_box = decoder_half1.vol_box//2 decoder_half2.vol_box = decoder_half2.vol_box//2 - gpu_box = decoder_half1.vol_box + decoder_half1.vol_box = decoder_half1.box_size decoder_half2.vol_box = decoder_half2.box_size if mask_file: