test cossim

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]``