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

tweak

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-31 18:19:43 +00:00
parent dbe35ee943
commit 172cf9d8ce
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
aimodel/src/lib/ai/components/LossContrastive.py
View file

@ -46,6 +46,10 @@ class LossContrastive(tf.keras.losses.Loss):
loss = (loss_rainfall + loss_water) / 2
print("LOSS", loss)
loss = tf.math.reduce_mean(loss)
print("LOSS FINAL", loss)
# cosine_similarity results in tensor of range -1 - 1, but tf.sparse.eye has range 0 - 1
# print("LABELS", labels)
# print("LOSS_rainfall", loss_rainfall)