mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-23 01:33:01 +00:00
tweak
This commit is contained in:
parent
dbe35ee943
commit
172cf9d8ce
1 changed files with 4 additions and 0 deletions
|
@ -46,6 +46,10 @@ class LossContrastive(tf.keras.losses.Loss):
|
||||||
loss = (loss_rainfall + loss_water) / 2
|
loss = (loss_rainfall + loss_water) / 2
|
||||||
|
|
||||||
print("LOSS", loss)
|
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
|
# cosine_similarity results in tensor of range -1 - 1, but tf.sparse.eye has range 0 - 1
|
||||||
# print("LABELS", labels)
|
# print("LABELS", labels)
|
||||||
# print("LOSS_rainfall", loss_rainfall)
|
# print("LOSS_rainfall", loss_rainfall)
|
||||||
|
|
Loading…
Reference in a new issue