diff --git a/model_custom.py b/model_custom.py index 6745da35457b1a067ac1e067ba1743f6f180f136..571dd3ad7f2be96959f5a6bc436e83635e3c4d3d 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)