Tingxie commited on
Commit
09b82aa
·
verified ·
1 Parent(s): 4bbef8d

Upload 2 files

Browse files
model/median_energy/checkpoints/config.yaml CHANGED
@@ -1,9 +1,9 @@
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_20/"
7
  log_every_n_steps: 2
8
  learning_rate: 1e-05
9
  weight_decay: 0.0001
@@ -25,8 +25,8 @@ 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_20.mgf"
29
- smi_file: "/home/xieting/cl_data/hcd3_0.01test/train_hcd_posh_20_smi.npy"
30
 
31
  loss:
32
  temperature: 0.1
 
1
  batch_size: 1000
2
+ epochs: 5000
3
  eval_every_n_epochs: 5
4
  smilesmodel_finetune: False
5
+ smilesmodel_finetune_path: ""
6
+ fine_tune_from: ""
7
  log_every_n_steps: 2
8
  learning_rate: 1e-05
9
  weight_decay: 0.0001
 
25
  s: 1
26
  num_workers: 0
27
  valid_size: 0.1
28
+ ms2_file: ""
29
+ smi_file: ""
30
 
31
  loss:
32
  temperature: 0.1
model/median_energy/checkpoints/model.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73bf22a8b3f0ad9a4cfbfa43b1a3e2efc40b6249d2c18d30269544649db0772a
3
+ size 21327230