mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-12-26 07:55:01 +00:00
fixup
This commit is contained in:
parent
b986b069e2
commit
5e60319024
1 changed files with 1 additions and 1 deletions
|
@ -61,4 +61,4 @@ if __name__ == "__main__":
|
|||
loss = LossContrastive(weight_temperature=weight_temperature, batch_size=64)
|
||||
|
||||
tensor_input = tf.random.uniform([64, 2, 512])
|
||||
print(loss(tensor_input))
|
||||
print(loss(tf.constant(1), tensor_input))
|
Loading…
Reference in a new issue