From 769eedb8718d8a52f15159515beeae4b2df882e4 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 21 Oct 2024 13:57:38 +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 82c1516..acc83b0 100644
--- a/model_custom.py
+++ b/model_custom.py
@@ -39,7 +39,7 @@ class PositionalEncoding(nn.Module):
class Model_Common_Transformer(nn.Module):
- def __init__(self, drop_rate=0.1, embedding_dim=128, nb_aa=21,
+ def __init__(self, drop_rate=0.1, embedding_dim=128, nb_aa=24,
regressor_layer_size_rt=512, regressor_layer_size_int=512, decoder_rt_ff=512, decoder_int_ff=512,
n_head=1, seq_length=25,
charge_max=4, charge_frag_max=3, encoder_ff=512, encoder_num_layer=1, decoder_rt_num_layer=1,
--
GitLab