sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 09:13: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

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

@ -15,6 +15,5 @@ class LayerConvNeXtGamma(tf.keras.layers.Layer):
def get_config(self):
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

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,
filepath=os.path.join(
dirpath_checkpoints,
"checkpoint_weights_e{epoch:d}_loss{loss:.3f}.hdf5"
"checkpoint_e{epoch:d}_loss{loss:.3f}.hdf5"
),
monitor="loss"
),