Upload 9 files
Browse files- N-Step-Checkpoint_0_0.ckpt +3 -0
- N-Step-Checkpoint_0_5000.ckpt +3 -0
- N-Step-Checkpoint_1_10000.ckpt +3 -0
- N-Step-Checkpoint_1_15000.ckpt +3 -0
- N-Step-Checkpoint_2_20000.ckpt +3 -0
- N-Step-Checkpoint_3_25000.ckpt +3 -0
- N-Step-Checkpoint_3_30000.ckpt +3 -0
- events.out.tfevents.1643396916.cccxc543.3427421.0 +3 -0
- hparams.yaml +68 -0
N-Step-Checkpoint_0_0.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1fe1ab7a82255112e004cb6f9395bc85b4be1bb5c7a3da3eacb82be99eff57aa
|
3 |
+
size 561703793
|
N-Step-Checkpoint_0_5000.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fe92d48032a75f47712a042553880e438e266b2bfebe6e2ca139a98c2a95d47f
|
3 |
+
size 561703921
|
N-Step-Checkpoint_1_10000.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:67c82822cc221459511a8148f920f1259f413bb40eada33fb59bb595e4c6a606
|
3 |
+
size 561703921
|
N-Step-Checkpoint_1_15000.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:a79add60526c7c3fb2d557fccb5f380059fe0fc4ad4e677ad9e9073298876a77
|
3 |
+
size 561703921
|
N-Step-Checkpoint_2_20000.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d6bb2bbe3657228972008335d4750142009f5098a81a6c214b2545fbd79f11ee
|
3 |
+
size 561703921
|
N-Step-Checkpoint_3_25000.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fd33c36da059c8c802db1a5168e4252c40d2e79f317b0406489e769785b78783
|
3 |
+
size 561703921
|
N-Step-Checkpoint_3_30000.ckpt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:60f0acac8e3cf7c6b521ab52e1e09756b81b48c977b97426be0f695961a38be5
|
3 |
+
size 561703921
|
events.out.tfevents.1643396916.cccxc543.3427421.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9f149ac34fa92d983c1ef475085baa416e1d4b98e9dee0b8b5216f4e94ef6348
|
3 |
+
size 296059
|
hparams.yaml
ADDED
@@ -0,0 +1,68 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
accelerator: ddp
|
2 |
+
batch_size: 64
|
3 |
+
beam_size: 0
|
4 |
+
checkpoint_every: 5000
|
5 |
+
clip_grad: 50
|
6 |
+
config_load: null
|
7 |
+
config_save: null
|
8 |
+
d_dropout: 0.2
|
9 |
+
data_path: ''
|
10 |
+
data_root: /dccstor/medscan7/smallmolecule/runs/ba-predictor/small-data/affinity
|
11 |
+
dataset_length: null
|
12 |
+
dataset_name: sol
|
13 |
+
debug: true
|
14 |
+
device: cuda
|
15 |
+
dropout: 0.1
|
16 |
+
eval_every: 1000
|
17 |
+
fast_dev_run: false
|
18 |
+
fc_h: 512
|
19 |
+
finetune_path: ''
|
20 |
+
freeze_model: false
|
21 |
+
from_scratch: false
|
22 |
+
gen_save: null
|
23 |
+
gpus: 8
|
24 |
+
grad_acc: 1
|
25 |
+
log_file: null
|
26 |
+
lr: 0.001
|
27 |
+
lr_end: 0.00030000000000000003
|
28 |
+
lr_multiplier: 8
|
29 |
+
lr_start: 3.0e-05
|
30 |
+
max_epochs: 4
|
31 |
+
max_len: 202
|
32 |
+
measure_name: measure
|
33 |
+
min_len: 1
|
34 |
+
mode: cls
|
35 |
+
model_arch: BERT_16GPU_Both_10percent_rotate_no_masking
|
36 |
+
model_load: null
|
37 |
+
model_save: model.pt
|
38 |
+
model_save_dir: ./models_dump/
|
39 |
+
n_batch: 1800
|
40 |
+
n_embd: 768
|
41 |
+
n_head: 12
|
42 |
+
n_jobs: 1
|
43 |
+
n_last: 1000
|
44 |
+
n_layer: 12
|
45 |
+
n_samples: null
|
46 |
+
n_workers: 8
|
47 |
+
nucleus_thresh: 0.9
|
48 |
+
num_epoch: 1
|
49 |
+
num_feats: 32
|
50 |
+
num_nodes: 1
|
51 |
+
num_seq_returned: 0
|
52 |
+
num_workers: 0
|
53 |
+
pretext_size: 0
|
54 |
+
q_dropout: 0.5
|
55 |
+
restart_path: ''
|
56 |
+
resume: false
|
57 |
+
root_dir: .
|
58 |
+
rotate: true
|
59 |
+
seed: 12345
|
60 |
+
smiles_embedding: /dccstor/medscan7/smallmolecule/runs/ba-predictor/small-data/embeddings/protein/ba_embeddings_tanh_512_2986138_2.pt
|
61 |
+
tb_loc: null
|
62 |
+
train_load: both
|
63 |
+
unlike_alpha: 1.0
|
64 |
+
unlikelihood: false
|
65 |
+
use_linear: 0
|
66 |
+
val_load: null
|
67 |
+
vocab_load: null
|
68 |
+
vocab_save: null
|