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

ai: add missing arguments to LossContrastive

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

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

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

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

@ -6,7 +6,7 @@ from .components.LayerContrastiveEncoder import LayerContrastiveEncoder
from .components.LayerCheeseMultipleOut import LayerCheeseMultipleOut
from .components.LossContrastive import LossContrastive
def model_rainfallwater_contrastive(shape_rainfall, shape_water, feature_dim=2048):
def model_rainfallwater_contrastive(shape_rainfall, shape_water, batch_size=64, feature_dim=2048):
logger.info(shape_rainfall)
logger.info(shape_water)
@ -48,5 +48,5 @@ def model_rainfallwater_contrastive(shape_rainfall, shape_water, feature_dim=204
model.compile(
optimizer="Adam",
loss=LossContrastive(weight_temperature=weight_temperature)
loss=LossContrastive(batch_size=batch_size, weight_temperature=weight_temperature)
)