Skip to content

Commit 2e5436f

Browse files
author
brettin
committed
for training a docking surrogate
1 parent 1356721 commit 2e5436f

File tree

1 file changed

+19
-0
lines changed

1 file changed

+19
-0
lines changed
Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,19 @@
1+
[Global_Params]
2+
data_url = "https://ftp.mcs.anl.gov/pub/candle/public/benchmarks/Examples/xform-smiles-data/"
3+
train_data = 'Nsp13.helicase_m1_pocket2.train'
4+
val_data = 'Nsp13.helicase_m1_pocket2.val'
5+
dropout = 0.1
6+
dense = [1024, 256, 64, 16]
7+
activation = 'relu'
8+
out_layer = 1
9+
out_activation = 'relu'
10+
transformer_depth = 4
11+
embed_dim = 128
12+
ff_dim = 128
13+
maxlen = 250
14+
num_heads = 16
15+
vocab_size = 40000
16+
epochs = 400
17+
batch_size = 32
18+
loss = 'mean_squared_error'
19+
optimizer='Adam'

0 commit comments

Comments
 (0)