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

View file

@ -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))