diff --git a/aimodel/src/lib/ai/components/convnext.py b/aimodel/src/lib/ai/components/convnext.py index b49fb38..dc3630a 100644 --- a/aimodel/src/lib/ai/components/convnext.py +++ b/aimodel/src/lib/ai/components/convnext.py @@ -147,7 +147,7 @@ def add_convnext_block(y, dim, drop_prob=0, prefix=""): name=f'{prefix}.pwconv1' )(y) - + y = tf.keras.layers.Activation( 'gelu', name=f'{prefix}.act'