From 26631234078adaeb2328b07cf733825b25a7226f Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 1 Sep 2022 17:40:23 +0100 Subject: [PATCH] moar logging --- aimodel/src/lib/ai/components/LayerContrastiveEncoder.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py b/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py index b5c2a3b..1b7bc48 100644 --- a/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py +++ b/aimodel/src/lib/ai/components/LayerContrastiveEncoder.py @@ -1,4 +1,5 @@ import tensorflow as tf +from loguru import logger # from tensorflow.keras.applications.resnet_v2 import ResNet50V2 from ..helpers.summarywriter import summarylogger @@ -21,6 +22,10 @@ class LayerContrastiveEncoder(tf.keras.layers.Layer): """ super(LayerContrastiveEncoder, self).__init__(**kwargs) + print(f"input_width: {input_width}") + print(f"input_height: {input_height}") + print(f"channels: {channels}") + self.param_input_width = input_width self.param_input_height = input_height self.param_channels = channels