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

mono: fix loading checkpoint

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-11-29 15:25:11 +00:00
parent dba6cbffcd
commit 838ff56a3b
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
aimodel/src/lib/ai/RainfallWaterMono.py
View file

@ -15,6 +15,7 @@ from .helpers import make_callbacks
from .helpers import summarywriter from .helpers import summarywriter
from .components.LayerConvNeXtGamma import LayerConvNeXtGamma from .components.LayerConvNeXtGamma import LayerConvNeXtGamma
from .components.LayerStack2Image import LayerStack2Image from .components.LayerStack2Image import LayerStack2Image
from .components.LossCrossentropy import LossCrossentropy
from .helpers.summarywriter import summarywriter from .helpers.summarywriter import summarywriter
class RainfallWaterMono(object): class RainfallWaterMono(object):
@ -71,7 +72,8 @@ class RainfallWaterMono(object):
self.model = tf.keras.models.load_model(filepath_checkpoint, custom_objects={ self.model = tf.keras.models.load_model(filepath_checkpoint, custom_objects={
"LayerConvNeXtGamma": LayerConvNeXtGamma, "LayerConvNeXtGamma": LayerConvNeXtGamma,
"LayerStack2Image": LayerStack2Image "LayerStack2Image": LayerStack2Image,
"LossCrossentropy": LossCrossentropy
}) })