sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-24 18:23:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

LayerConvNeXtGamma: fix for mixed precision mode

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-02-02 16:22:08 +00:00
parent a630db2c49
commit 1a8f10339a
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/lib/ai/components/LayerConvNeXtGamma.py
View file

@ -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))
self.const = const_val * tf.ones((self.dim), dtype=tf.float32 if tf.mixed_precision.global_policy().name == "float32" else tf.float16)
def call(self, inputs, **kwargs):
return tf.multiply(inputs, self.const)