propagate summary definitions

ctu-geoforall-lab-projects · Sep 21, 2023 · 7e749d3 · 7e749d3
1 parent fa3e095
commit 7e749d3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/architectures.py b/src/architectures.py
@@ -136,7 +136,7 @@ def summary(self, line_length=None, positions=None, print_fn=None):
                         self.nr_bands), name='input')
         model = Model(inputs=[inputs], outputs=self.call(inputs),
                       name=self.name)
-        return model.summary()
+        return model.summary(line_length, positions, print_fn)
 
     @abstractmethod
     def instantiate_layers(self):