katielink commited on
Commit
aebb2d9
1 Parent(s): 653ba71

Upload 9 files

Browse files
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