From 103c370d3faf1178e824c350bcad126049b68d55 Mon Sep 17 00:00:00 2001 From: lukasmahieu Date: Sun, 16 Jun 2024 14:26:07 +0200 Subject: [PATCH] Fix basenji model parameter name --- src/crested/tl/zoo/utils/_layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/crested/tl/zoo/utils/_layers.py b/src/crested/tl/zoo/utils/_layers.py index 695cc7d..5dbd4ca 100644 --- a/src/crested/tl/zoo/utils/_layers.py +++ b/src/crested/tl/zoo/utils/_layers.py @@ -297,7 +297,7 @@ def conv_block_bs( use_bias=False, dilation_rate=dilation_rate, kernel_initializer=kernel_initializer, - l2=tf.keras.regularizers.l2(l2_scale), + kernel_regularizer=tf.keras.regularizers.l2(l2_scale), )(current) # batch norm