From d8d2f555fb26055b3672ec0a86444cf00b63f62e Mon Sep 17 00:00:00 2001 From: Schneider Leo <leo.schneider@etu.ec-lyon.fr> Date: Mon, 21 Oct 2024 10:04:55 +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 6745da3..571dd3a 100644 --- a/model_custom.py +++ b/model_custom.py @@ -102,7 +102,7 @@ class Model_Common_Transformer(nn.Module): def forward(self, seq, charge): 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() - print(seq_emb.shape()) + print(seq_emb.shape) print(self.nb_aa, self.embedding_dim) emb = self.pos_embedding(self.emb(seq_emb)) meta_enc = self.meta_enc(meta_ohe) -- GitLab