From a2e73d412b2c77faf68ffcda30640715018297e9 Mon Sep 17 00:00:00 2001
From: lschneider <leo.schneider@univ-lyon1.fr>
Date: Thu, 19 Sep 2024 14:59:34 +0200
Subject: [PATCH] test cossim
---
model_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/model_custom.py b/model_custom.py
index 0a67502..b473a04 100644
--- a/model_custom.py
+++ b/model_custom.py
@@ -30,7 +30,7 @@ class PositionalEncoding(nn.Module):
def forward(self, x):
- x = torch.permute(x, (1, 0, 2))
+
"""
Arguments:
x: Tensor, shape ``[seq_len, batch_size, embedding_dim]``
--
GitLab