From 5e60319024325c284adbcd9fc5398507c77841d0 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Mon, 31 Oct 2022 17:56:49 +0000 Subject: [PATCH] fixup --- aimodel/src/lib/ai/components/LossContrastive.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/lib/ai/components/LossContrastive.py b/aimodel/src/lib/ai/components/LossContrastive.py index d54cf62..c656c96 100644 --- a/aimodel/src/lib/ai/components/LossContrastive.py +++ b/aimodel/src/lib/ai/components/LossContrastive.py @@ -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)) \ No newline at end of file + print(loss(tf.constant(1), tensor_input)) \ No newline at end of file