From 458faa96d2adb803444e4ed5b173d3357902b001 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 31 Oct 2022 17:18:21 +0000 Subject: [PATCH] loss: fixup --- aimodel/src/lib/ai/components/LossContrastive.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/components/LossContrastive.py b/aimodel/src/lib/ai/components/LossContrastive.py index 59a2f8f..356147f 100644 --- a/aimodel/src/lib/ai/components/LossContrastive.py +++ b/aimodel/src/lib/ai/components/LossContrastive.py @@ -19,7 +19,7 @@ class LossContrastive(tf.keras.losses.Loss): # normalise features rainfall = rainfall / tf.math.l2_normalize(rainfall, axis=1) - rainfall = rainfall / tf.math.l2_normalize(rainfall, axis=1) + water = water / tf.math.l2_normalize(water, axis=1) # logits = tf.linalg.matmul(rainfall, tf.transpose(water)) * tf.clip_by_value(tf.math.exp(self.weight_temperature), 0, 100) logits = tf.linalg.matmul(rainfall, tf.transpose(water)) * tf.math.exp(self.weight_temperature)