fix: name in add_weight()

awsaf49 · Dec 24, 2023 · e2176f1 · e2176f1
1 parent 1473a56
commit e2176f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gcvit/layers/block.py b/gcvit/layers/block.py
@@ -65,14 +65,14 @@ def build(self, input_shape):
         )
         if self.layer_scale is not None:
             self.gamma1 = self.add_weight(
-                "gamma1",
+                name="gamma1",
                 shape=[C],
                 initializer=tf.keras.initializers.Constant(self.layer_scale),
                 trainable=True,
                 dtype=self.dtype,
             )
             self.gamma2 = self.add_weight(
-                "gamma2",
+                name="gamma2",
                 shape=[C],
                 initializer=tf.keras.initializers.Constant(self.layer_scale),
                 trainable=True,