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

fixup

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-31 17:56:49 +00:00
parent b986b069e2
commit 5e60319024
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -61,4 +61,4 @@ if __name__ == "__main__":
loss = LossContrastive(weight_temperature=weight_temperature, batch_size=64) loss = LossContrastive(weight_temperature=weight_temperature, batch_size=64)
tensor_input = tf.random.uniform([64, 2, 512]) tensor_input = tf.random.uniform([64, 2, 512])
print(loss(tensor_input)) print(loss(tf.constant(1), tensor_input))