sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-25 10:32:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

LayerConvNextGamma: fix config serialisation bug

Browse source 
.....this is unlikely to be the problem as this bug is in an unused code path.
This commit is contained in:
Starbeamrainbowlabs 2022-11-25 21:16:31 +00:00
parent e7410fb480
commit f39e4ade70
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

@ -15,5 +15,5 @@ class LayerConvNeXtGamma(tf.keras.layers.Layer):
def get_config(self): def get_config(self):
config = super(LayerConvNeXtGamma, self).get_config() config = super(LayerConvNeXtGamma, self).get_config()
config.update({ "const": self.const.numpy(), "dim": self.dim }) config.update({ "const_val": self.const.numpy(), "dim": self.dim })
return config return config