From 530134011fb8a5ce52a4b232da81082240b446a8 Mon Sep 17 00:00:00 2001 From: Enrique Mondragon Estrada <101150878+enriquemondragon@users.noreply.github.com> Date: Fri, 28 Feb 2025 21:11:01 -0500 Subject: [PATCH] Update layers shape in seqnn.py --- basenji/seqnn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/basenji/seqnn.py b/basenji/seqnn.py index 11f57ddc..f8b3d355 100644 --- a/basenji/seqnn.py +++ b/basenji/seqnn.py @@ -187,7 +187,7 @@ def build_model(self, save_reprs=True): self.model_strides.append(1) for layer in self.model.layers: if hasattr(layer, 'strides') or hasattr(layer, 'size'): - stride_factor = layer.input_shape[1] / layer.output_shape[1] + stride_factor = layer.input.shape[1] / layer.output.shape[1] self.model_strides[-1] *= stride_factor self.model_strides[-1] = int(self.model_strides[-1])