Hugging Face's logo

cyclization9
/
SPURS

Model card Files
xet
 Community
cyclization9 commited on
Commit
3dada1f
·
verified ·
1 Parent(s): 9ff476a

Update SPURS_multi checkpoints

Browse files
Files changed (5) hide show
  1. spurs_multi/.hydra/config.yaml +1 -1
  2. spurs_multi/config_tree.log +118 -0
  3. spurs_multi/multi_additive_before_train_result_25.9.25.pt +3 -0
  4. spurs_multi/multi_additive_result_25.9.25.pt +3 -0
  5. spurs_multi/multi_result_25.9.25.pt +3 -0
spurs_multi/.hydra/config.yaml CHANGED
@@ -37,7 +37,7 @@ model:
37
  flat_dim: 128
38
  input_dim: 256
39
  hidden_dim: 1024
40
- num_layers: 2
41
  output_dim: 21
42
  dropout: 0.1
43
  multi:
 
37
  flat_dim: 128
38
  input_dim: 256
39
  hidden_dim: 1024
40
+ num_layers: 4
41
  output_dim: 21
42
  dropout: 0.1
43
  multi:
spurs_multi/config_tree.log ADDED
@@ -0,0 +1,118 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ CONFIG
2
+ ├── datamodule
3
+ │ └── _target_: megascale_multi
4
+ │ batch_size: 1
5
+ │ num_workers: 8
6
+ │ pin_memory: true
7
+ │ alphabet:
8
+ │ name: esm
9
+ │ featurizer: megascale
10
+ │
11
+ ├── task
12
+ │ └── _target_: stability/megascale
13
+ │ criterion:
14
+ │ _target_: spurs.modules.mseloss.MSELoss
15
+ │ optimizer:
16
+ │ type: adamw
17
+ │ _partial_: true
18
+ │ lr: 0.0001
19
+ │ betas:
20
+ │ - 0.9
21
+ │ - 0.98
22
+ │ weight_decay: 0.0001
23
+ │ lr_scheduler:
24
+ │ type: plateau
25
+ │ lr: 0.001
26
+ │ factor: 0.2
27
+ │ mode: min
28
+ │ monitor: val/rho_avg_mse
29
+ │
30
+ ├── model
31
+ │ └── _target_: spurs_multi
32
+ │ encoder:
33
+ │ d_model: 128
34
+ │ n_enc_layers: 3
35
+ │ n_dec_layers: 3
36
+ │ tune: true
37
+ │ use_input_decoding_order: true
38
+ │ adapter_layer_indices:
39
+ │ - -3
40
+ │ mlp:
41
+ │ flat_dim: 128
42
+ │ input_dim: 256
43
+ │ hidden_dim: 1024
44
+ │ num_layers: 2
45
+ │ output_dim: 21
46
+ │ dropout: 0.1
47
+ │ multi:
48
+ │ flat_dim: 128
49
+ │ input_dim: 384
50
+ │ hidden_dim: 768
51
+ │ num_layers: 4
52
+ │ output_dim: 1
53
+ │ dropout: 0.1
54
+ │ dropout: 0.5
55
+ │ agg_type: concate
56
+ │ agg_dim: 128
57
+ │
58
+ ├── callbacks
59
+ │ └── model_summary:
60
+ │ _target_: pytorch_lightning.callbacks.RichModelSummary
61
+ │ max_depth: -1
62
+ │ model_checkpoint:
63
+ │ _target_: spurs.utils.callbacks.ModelCheckpoint
64
+ │ monitor: val/rho_avg_spearman
65
+ │ mode: max
66
+ │ save_top_k: 1
67
+ │ save_last: true
68
+ │ verbose: true
69
+ │ dirpath: checkpoints
70
+ │ filename: step_{global_step}-val/rho_avg_spearman_{val/rho_avg_spearman:.2f}
71
+ │ auto_insert_metric_name: false
72
+ │ early_stopping:
73
+ │ _target_: pytorch_lightning.callbacks.EarlyStopping
74
+ │ monitor: val/rho_avg_spearman
75
+ │ mode: max
76
+ │ patience: 10
77
+ │ min_delta: 0
78
+ │ check_on_train_epoch_end: false
79
+ │
80
+ ├── trainer
81
+ │ └── _target_: pytorch_lightning.Trainer
82
+ │ gpus: auto
83
+ │ min_epochs: 1
84
+ │ max_epochs: 10
85
+ │ enable_progress_bar: true
86
+ │ log_every_n_steps: 10
87
+ │ resume_from_checkpoint: null
88
+ │
89
+ ├── experiment_path
90
+ │ └── /nethome/zli3161/DATA-nash/SPURS/data/checkpoints/spurs_multi
91
+ ├── ckpt_path
92
+ │ └── best.ckpt
93
+ ├── data_split
94
+ │ └── test
95
+ ├── mode
96
+ │ └── predict
97
+ ├── name
98
+ │ └── megascale/spurs_multi
99
+ ├── train
100
+ │ └── seed: 49
101
+ │ lr: 0.001
102
+ │ monitor: val/rho_avg_spearman
103
+ │ mode: max
104
+ │ patience: 10
105
+ │
106
+ ├── print_config
107
+ │ └── True
108
+ ├── ignore_warnings
109
+ │ └── True
110
+ ├── seed
111
+ │ └── 42
112
+ └── paths
113
+ └── root_dir: /work/ziang/SPURS
114
+ data_dir: /work/ziang/SPURS/data/
115
+ log_dir: /nethome/zli3161/DATA-nash/SPURS/data/checkpoints/spurs_multi
116
+ ckpt_dir: /nethome/zli3161/DATA-nash/SPURS/data/checkpoints/spurs_multi/checkpoints
117
+ psnetdata: /work/ziang/SPURS/PSNet/data
118
+
spurs_multi/multi_additive_before_train_result_25.9.25.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0e7b742b64a73b2591c8c3e8e8c4bd84ed64f90b2d1cca722f62a2339041416b
3
+ size 1237735
spurs_multi/multi_additive_result_25.9.25.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b651f102e1d9ad7b2dca84d29c50e9e8cdb593d0b9d2e18a334b40beb8f3254e
3
+ size 1237735
spurs_multi/multi_result_25.9.25.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d68f86bb83b1a81923c5583132980ae26ebc9ad8448e7396dc965f5cbbdb4f60
3
+ size 1237735