From 5846828f9ef68d2f7dce177a2b18e916d2cb2342 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 31 Aug 2022 17:48:09 +0100 Subject: [PATCH] debug logging --- aimodel/src/lib/ai/model_rainfallwater_contrastive.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py index 2bff097..f7af3a7 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_contrastive.py +++ b/aimodel/src/lib/ai/model_rainfallwater_contrastive.py @@ -1,4 +1,5 @@ -from pickletools import optimize + +from loguru import logger import tensorflow as tf from .components.LayerContrastiveEncoder import LayerContrastiveEncoder @@ -6,6 +7,8 @@ from .components.LayerCheeseMultipleOut import LayerCheeseMultipleOut from .components.LossContrastive import LossContrastive def model_rainfallwater_contrastive(shape_rainfall, shape_water, feature_dim=200): + logger.info(shape_rainfall) + logger.info(shape_water) rainfall_width, rainfall_height, rainfall_channels = shape_rainfall water_width, water_height, water_channels = shape_water