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

ai: set self.model

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-09-01 16:20:23 +01:00
parent f1be5fe2bd
commit 940f7aa1b5
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 4 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
aimodel/src/lib/ai/RainfallWaterContraster.py
View file

@ -57,8 +57,7 @@ class RainfallWaterContraster(object):
def make_model(self): def make_model(self):
model = model_rainfallwater_contrastive(batch_size=self.batch_size, **self.kwargs) return model_rainfallwater_contrastive(batch_size=self.batch_size, **self.kwargs)
return model
def load_model(self, filepath_checkpoint): def load_model(self, filepath_checkpoint):

2
aimodel/src/lib/ai/model_rainfallwater_contrastive.py
View file

@ -50,3 +50,5 @@ def model_rainfallwater_contrastive(shape_rainfall, shape_water, batch_size=64,
optimizer="Adam", optimizer="Adam",
loss=LossContrastive(batch_size=batch_size, weight_temperature=weight_temperature) loss=LossContrastive(batch_size=batch_size, weight_temperature=weight_temperature)
) )
return model