From 8446a842d1d81aa07aa2624706ea81d4b987d2d3 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Fri, 3 Feb 2023 16:01:54 +0000 Subject: [PATCH] typo --- aimodel/src/lib/ai/components/LayerConvNeXtGamma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/components/LayerConvNeXtGamma.py b/aimodel/src/lib/ai/components/LayerConvNeXtGamma.py index 1189de5..c3b75da 100644 --- a/aimodel/src/lib/ai/components/LayerConvNeXtGamma.py +++ b/aimodel/src/lib/ai/components/LayerConvNeXtGamma.py @@ -7,7 +7,7 @@ class LayerConvNeXtGamma(tf.keras.layers.Layer): super(LayerConvNeXtGamma, self).__init__(name=name) self.dim = dim - self.const = const_val * tf.ones((self.dim), dtype=tf.float32 if tf.mixed_precision.global_policy().name == "float32" else tf.float16) + self.const = const_val * tf.ones((self.dim), dtype=tf.float32 if tf.keras.mixed_precision.global_policy().name == "float32" else tf.float16) def call(self, inputs, **kwargs): return tf.multiply(inputs, self.const)