[#51686] modelwrappers: frameworks: tensorflow: Fixed printing the model

Signed-off-by: Grzegorz Latosinski <[email protected]>
antmicro · Nov 27, 2023 · 8f0b905 · 8f0b905
1 parent a433e71
commit 8f0b905
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kenning/modelwrappers/frameworks/tensorflow.py b/kenning/modelwrappers/frameworks/tensorflow.py
@@ -54,7 +54,7 @@ def load_model(self, model_path: PathOrURI):
         if hasattr(self, "model") and self.model is not None:
             del self.model
         self.model = tf.keras.models.load_model(str(model_path))
-        print(self.model.summary())
+        self.model.summary()
 
     def save_model(self, model_path: PathOrURI):
         self.prepare_model()