From 53dfa3268592578bab2ebed6859d8e317c876434 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 1 Dec 2022 15:10:51 +0000 Subject: [PATCH] model_mono: log learning rate --- aimodel/src/lib/ai/model_rainfallwater_mono.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/aimodel/src/lib/ai/model_rainfallwater_mono.py b/aimodel/src/lib/ai/model_rainfallwater_mono.py index 5ab3471..45ba01d 100644 --- a/aimodel/src/lib/ai/model_rainfallwater_mono.py +++ b/aimodel/src/lib/ai/model_rainfallwater_mono.py @@ -73,6 +73,8 @@ def model_rainfallwater_mono(metadata, shape_water_out, model_arch_enc="convnext outputs = layer_next ) + logger.info(f"learning_rate: {str(learning_rate)}") + optimizer = "Adam" if learning_rate is not None: optimizer = tf.keras.optimizers.Adam(learning_rate=learning_rate)