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

test

parent 808c2aa6
No related branches found
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
dim_encoder = 768
[MODEL_LINKER]
nhead=8
nhead=16
dim_emb_atom = 256
dim_feedforward_transformer = 512
num_layers=3
......@@ -24,7 +24,7 @@ dropout=0.1
sinkhorn_iters=5
[MODEL_TRAINING]
batch_size=32
batch_size=16
epoch=30
seed_val=42
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