mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-25 10:32:59 +00:00
model_mono: log learning rate
This commit is contained in:
parent
c384d55dff
commit
53dfa32685
1 changed files with 2 additions and 0 deletions
|
@ -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)
|
||||
|
|
Loading…
Reference in a new issue