From c1ca0fcc52b5c81ed4dd5cb670d98cb458afedb2 Mon Sep 17 00:00:00 2001
From: schne <leo.schneider@ecl19.ec-lyon.fr>
Date: Tue, 20 Aug 2024 17:09:25 +0200
Subject: [PATCH] main cutom on ISA data
---
model_custom.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/model_custom.py b/model_custom.py
index 96cfc64..4e245e2 100644
--- a/model_custom.py
+++ b/model_custom.py
@@ -101,6 +101,7 @@ 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