From f92b2b34722a897b177ee873647ba7c00047fb47 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 2 Dec 2022 17:22:46 +0000 Subject: [PATCH] according to the equation it looks like it's 2 --- aimodel/src/lib/ai/components/cbam.py | 1 - 1 file changed, 1 deletion(-) diff --git a/aimodel/src/lib/ai/components/cbam.py b/aimodel/src/lib/ai/components/cbam.py index cfdedeb..ba85684 100644 --- a/aimodel/src/lib/ai/components/cbam.py +++ b/aimodel/src/lib/ai/components/cbam.py @@ -36,7 +36,6 @@ class LayerCBAMAttentionChannel(tf.keras.layers.Layer): self.param_reduction_ratio = reduction_ratio self.mlp = tf.keras.Sequential([ - tf.keras.layers.Dense(self.param_dim), tf.keras.layers.Dense(self.param_dim / self.param_reduction_ratio), tf.keras.layers.Dense(self.param_dim) ])