From 1de23a4aae3f07ae01f8c245f2b2972c2b58fb09 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 21 Oct 2024 13:17:59 +0200
Subject: [PATCH] datasets
---
model_custom.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/model_custom.py b/model_custom.py
index 04f0870..f829002 100644
--- a/model_custom.py
+++ b/model_custom.py
@@ -106,7 +106,7 @@ class Model_Common_Transformer(nn.Module):
seq_emb = torch.nn.functional.one_hot(seq, self.nb_aa).float()
print(seq_emb.shape)
print(self.nb_aa, self.embedding_dim)
- print(self.emb(seq_emb))
+ print(seq_emb)
emb = self.pos_embedding(self.emb(seq_emb))
meta_enc = self.meta_enc(meta_ohe)
--
GitLab