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