model/low_energy/checkpoints/config.yaml DELETED
@@ -1,34 +0,0 @@
1
- batch_size: 1000
2
- epochs: 3500
3
- eval_every_n_epochs: 5
4
- smilesmodel_finetune: False
5
- smilesmodel_finetune_path: "/public/home/hpc212307003/molclr_cnn/nist_w2v/pretrained_gin/checkpoints/model.pth"
6
- fine_tune_from: "/home/xieting/graph_transformer_esa/runs/dem4_esa_4.5_10/"
7
- log_every_n_steps: 2
8
- learning_rate: 1e-05
9
- weight_decay: 0.0001
10
- fp16_precision: True
11
- truncation: True
12
-
13
- model_config:
14
- num_layer: 5
15
- emb_dim: 300
16
- feat_dim: 512
17
- drop_ratio: 0.3
18
- pool : mean
19
- spec_embed_dim: 256
20
- dropout: 0.1
21
- layers: 3
22
- embed_dim: 256
23
-
24
- dataset:
25
- s: 1
26
- num_workers: 0
27
- valid_size: 0.1
28
- ms2_file: "/home/xieting/cl_data/hcd3_0.01test/train_hcd_posh_10.mgf"
29
- smi_file: "/home/xieting/cl_data/hcd3_0.01test/train_hcd_posh_10_smi.npy"
30
-
31
- loss:
32
- temperature: 0.1
33
- use_cosine_similarity: True
34
- alpha_weight: 0.75