ai: set self.model

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

View file

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

View file

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