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

LayerConvNeXtGamma: avoid adding an EagerTensor to config

Browse source 
Very weird how this is a problem when it wasn't before..
This commit is contained in:
Starbeamrainbowlabs 2022-10-12 17:12:07 +01:00
parent 32f5200d3b
commit 6423bf6702
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 14 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -3,18 +3,17 @@ import tensorflow as tf
# Code from https://github.com/leanderme/ConvNeXt-Tensorflow/blob/main/ConvNeXt.ipynb # Code from https://github.com/leanderme/ConvNeXt-Tensorflow/blob/main/ConvNeXt.ipynb
class LayerConvNeXtGamma(tf.keras.layers.Layer): class LayerConvNeXtGamma(tf.keras.layers.Layer):
def __init__(self, const_val = 1e-6, dim = None, name=None, **kwargs): def __init__(self, const_val = 1e-6, dim = None, name=None, **kwargs):
super(LayerConvNeXtGamma, self).__init__(name=name) super(LayerConvNeXtGamma, self).__init__(name=name)
self.dim = dim self.dim = dim
self.const = const_val * tf.ones((self.dim)) self.const = const_val * tf.ones((self.dim))
def call(self, inputs, **kwargs): def call(self, inputs, **kwargs):
return tf.multiply(inputs, self.const) return tf.multiply(inputs, self.const)
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, "dim": self.dim }) config.update({ "const": self.const.numpy(), "dim": self.dim })
return config
return config

2
aimodel/src/lib/ai/helpers/make_callbacks.py
View file

@ -16,7 +16,7 @@ def make_callbacks(dirpath, model_predict):
model_to_checkpoint=model_predict, model_to_checkpoint=model_predict,
filepath=os.path.join( filepath=os.path.join(
dirpath_checkpoints, dirpath_checkpoints,
"checkpoint_weights_e{epoch:d}_loss{loss:.3f}.hdf5" "checkpoint_e{epoch:d}_loss{loss:.3f}.hdf5"
), ),
monitor="loss" monitor="loss"
), ),