From b39e283669ba6504d3dbbca36664f2fa948c57d0 Mon Sep 17 00:00:00 2001 From: rhoadesScholar Date: Thu, 9 Jan 2025 16:55:23 -0500 Subject: [PATCH] =?UTF-8?q?fix:=20=F0=9F=90=9B=20Ensure=20input=20shapes?= =?UTF-8?q?=20are=20integers=20in=20Architecture=20class?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/cellmap_models/pytorch/cosem/load_model.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/cellmap_models/pytorch/cosem/load_model.py b/src/cellmap_models/pytorch/cosem/load_model.py index 1857087..05f52df 100755 --- a/src/cellmap_models/pytorch/cosem/load_model.py +++ b/src/cellmap_models/pytorch/cosem/load_model.py @@ -250,9 +250,9 @@ def compute_minimal_shapes(self): ) * np.prod(self.downsample_factors[lv:], axis=0) min_output_shape -= total_pad - self.min_input_shape = min_input_shape - self.min_output_shape = min_output_shape - self.input_size_step = step + self.min_input_shape = [int(s) for s in min_input_shape] + self.min_output_shape = [int(s) for s in min_output_shape] + self.input_size_step = int(step) def round_to_valid_input_shape(self, shape=None, mode="grow"): """