Fix basenji model parameter name

aertslab · Jun 16, 2024 · 103c370 · 103c370
1 parent ee9b614
commit 103c370
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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