Skip to content
Snippets Groups Projects
Commit c5851000 authored by Caroline DE POURTALES's avatar Caroline DE POURTALES
Browse files

test

parent 808c2aa6
Branches
No related tags found
3 merge requests!6Linker with transformer,!5Linker with transformer,!3Working on padding
...@@ -12,7 +12,7 @@ max_atoms_in_one_type=324 ...@@ -12,7 +12,7 @@ max_atoms_in_one_type=324
dim_encoder = 768 dim_encoder = 768
[MODEL_LINKER] [MODEL_LINKER]
nhead=8 nhead=16
dim_emb_atom = 256 dim_emb_atom = 256
dim_feedforward_transformer = 512 dim_feedforward_transformer = 512
num_layers=3 num_layers=3
...@@ -24,7 +24,7 @@ dropout=0.1 ...@@ -24,7 +24,7 @@ dropout=0.1
sinkhorn_iters=5 sinkhorn_iters=5
[MODEL_TRAINING] [MODEL_TRAINING]
batch_size=32 batch_size=16
epoch=30 epoch=30
seed_val=42 seed_val=42
learning_rate=2e-3 learning_rate=2e-3
\ No newline at end of file
File added
File added
File added
File added
File added
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment