ai: add missing arguments to LossContrastive

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

View file

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

View file

@ -6,7 +6,7 @@ from .components.LayerContrastiveEncoder import LayerContrastiveEncoder
from .components.LayerCheeseMultipleOut import LayerCheeseMultipleOut from .components.LayerCheeseMultipleOut import LayerCheeseMultipleOut
from .components.LossContrastive import LossContrastive 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_rainfall)
logger.info(shape_water) logger.info(shape_water)
@ -48,5 +48,5 @@ def model_rainfallwater_contrastive(shape_rainfall, shape_water, feature_dim=204
model.compile( model.compile(
optimizer="Adam", optimizer="Adam",
loss=LossContrastive(weight_temperature=weight_temperature) loss=LossContrastive(batch_size=batch_size, weight_temperature=weight_temperature)
) )