From f76c4b0f2d7d263340cdd6dba54fe4a428ef0bdd Mon Sep 17 00:00:00 2001
From: schne <leo.schneider@ecl19.ec-lyon.fr>
Date: Wed, 21 Aug 2024 10:53:14 +0200
Subject: [PATCH] main custom on ISA data
---
model_custom.py | 1 -
1 file changed, 1 deletion(-)
diff --git a/model_custom.py b/model_custom.py
index 4e245e2..96cfc64 100644
--- a/model_custom.py
+++ b/model_custom.py
@@ -101,7 +101,6 @@ class Model_Common_Transformer(nn.Module):
d_model=self.embedding_dim)
def forward(self, seq, charge):
- print(seq.shape)
meta_ohe = torch.nn.functional.one_hot(charge - 1, self.charge_max).float()
seq_emb = torch.nn.functional.one_hot(seq, self.nb_aa).float()
emb = self.pos_embedding(self.emb(seq_emb))
--
GitLab