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

loss: comment l2 norm

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-31 18:09:03 +00:00
parent 5e60319024
commit dbe35ee943
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 2 additions and 2 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

@ -20,8 +20,8 @@ class LossContrastive(tf.keras.losses.Loss):
# water = tf.reshape(water, [self.batch_size, water.shape[1]]) # water = tf.reshape(water, [self.batch_size, water.shape[1]])
# normalise features # 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) # water = water / tf.math.l2_normalize(water, axis=1)
print("AFTER_L2 rainfall", rainfall) print("AFTER_L2 rainfall", rainfall)
print("AFTER_L2 water", water) print("AFTER_L2 water", water)