From 0b8928dcafe9eeb5f84c35b24ff0bf719d75ea2e Mon Sep 17 00:00:00 2001 From: Carlos Tseng Date: Mon, 5 Aug 2024 16:56:54 +0800 Subject: [PATCH] Fix issue#256 --- tcn/tcn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tcn/tcn.py b/tcn/tcn.py index 0de9f89..a0579a1 100644 --- a/tcn/tcn.py +++ b/tcn/tcn.py @@ -315,7 +315,7 @@ def build(self, input_shape): else: self.output_slice_index = -1 # causal case. self.slicer_layer = Lambda(lambda tt: tt[:, self.output_slice_index, :], name='Slice_Output') - self.slicer_layer.build(self.build_output_shape.as_list()) + self.slicer_layer.build(list(self.build_output_shape)) def compute_output_shape(self, input_shape): """