ribesstefano commited on
Commit
22d1b5d
·
verified ·
1 Parent(s): 78f5736

Training in progress, step 80000, checkpoint

Browse files
last-checkpoint/generation_config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "_from_model_config": true,
3
  "bos_token_id": 0,
4
  "do_sample": true,
5
  "eos_token_id": 2,
 
1
  {
 
2
  "bos_token_id": 0,
3
  "do_sample": true,
4
  "eos_token_id": 2,
last-checkpoint/model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:57ccebdd082ea318372ffeb90a683283c04a1d53aa3b54c67fa99060a8ac05eb
3
  size 409608164
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:629c59066387572d7186cec85f68fd09a29679e9ced77cb7e2d02555652617c3
3
  size 409608164
last-checkpoint/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6147e34482389f722b0f2bd04dd98805d253ab94e45aaad38f239fa78576356a
3
  size 814647162
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5be61bd1ed3b0487618e6ef35be00a666572bd075826e9a7ece4bc83a9715dda
3
  size 814647162
last-checkpoint/rng_state.pth CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ba80147f178822473ec973474efa5be9a54273706827c8cc8ddf6b54fd92f4b6
3
  size 14244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:30741782ce3e29819a20fc22da36ffe445b9029830f9da81defafd4f5b91b4ab
3
  size 14244
last-checkpoint/scheduler.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ec08642f74697f111f64e755e2fc9bd211b07cc951ee155be7841c13ee3a6fae
3
  size 1192
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ee57e0668b261d4038c7356e1cb6d8eb8143fd9b8797c75ac5ac18791da56331
3
  size 1192
last-checkpoint/trainer_state.json CHANGED
@@ -1,9 +1,9 @@
1
  {
2
- "best_metric": null,
3
- "best_model_checkpoint": null,
4
- "epoch": 23.668639053254438,
5
  "eval_steps": 20000,
6
- "global_step": 60000,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
@@ -91,6 +91,76 @@
91
  "learning_rate": 5e-05,
92
  "loss": 0.0007,
93
  "step": 60000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
94
  }
95
  ],
96
  "logging_steps": 5000,
@@ -110,7 +180,7 @@
110
  "attributes": {}
111
  }
112
  },
113
- "total_flos": 5.1811282772976576e+17,
114
  "train_batch_size": 128,
115
  "trial_name": null,
116
  "trial_params": null
 
1
  {
2
+ "best_metric": 0.5856055240793201,
3
+ "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_reduce-rand-smiles-train-0.25/checkpoint-80000",
4
+ "epoch": 31.558185404339252,
5
  "eval_steps": 20000,
6
+ "global_step": 80000,
7
  "is_hyper_param_search": false,
8
  "is_local_process_zero": true,
9
  "is_world_process_zero": true,
 
91
  "learning_rate": 5e-05,
92
  "loss": 0.0007,
93
  "step": 60000
94
+ },
95
+ {
96
+ "epoch": 25.641025641025642,
97
+ "grad_norm": 0.051321662962436676,
98
+ "learning_rate": 5e-05,
99
+ "loss": 0.0006,
100
+ "step": 65000
101
+ },
102
+ {
103
+ "epoch": 27.613412228796843,
104
+ "grad_norm": 0.016928382217884064,
105
+ "learning_rate": 5e-05,
106
+ "loss": 0.0006,
107
+ "step": 70000
108
+ },
109
+ {
110
+ "epoch": 29.585798816568047,
111
+ "grad_norm": 0.02291979268193245,
112
+ "learning_rate": 5e-05,
113
+ "loss": 0.0006,
114
+ "step": 75000
115
+ },
116
+ {
117
+ "epoch": 31.558185404339252,
118
+ "grad_norm": 0.03006165847182274,
119
+ "learning_rate": 5e-05,
120
+ "loss": 0.0006,
121
+ "step": 80000
122
+ },
123
+ {
124
+ "epoch": 31.558185404339252,
125
+ "eval_all_ligands_equal": 0.5856055240793201,
126
+ "eval_e3_equal": 0.8271954674220963,
127
+ "eval_e3_graph_edit_distance": Infinity,
128
+ "eval_e3_graph_edit_distance_norm": Infinity,
129
+ "eval_e3_has_attachment_point(s)": 0.9770715297450425,
130
+ "eval_e3_heavy_atoms_difference": 0.8041784702549575,
131
+ "eval_e3_heavy_atoms_difference_norm": 0.02507045406737937,
132
+ "eval_e3_tanimoto_similarity": 0.0,
133
+ "eval_e3_valid": 0.9770715297450425,
134
+ "eval_has_all_attachment_points": 0.9890226628895185,
135
+ "eval_has_three_substructures": 0.9999114730878187,
136
+ "eval_heavy_atoms_difference": 5.752301699716714,
137
+ "eval_heavy_atoms_difference_norm": 0.0757990141580953,
138
+ "eval_linker_equal": 0.8324185552407932,
139
+ "eval_linker_graph_edit_distance": Infinity,
140
+ "eval_linker_graph_edit_distance_norm": Infinity,
141
+ "eval_linker_has_attachment_point(s)": 0.9967245042492918,
142
+ "eval_linker_heavy_atoms_difference": 0.2646954674220963,
143
+ "eval_linker_heavy_atoms_difference_norm": 0.005140908527503901,
144
+ "eval_linker_tanimoto_similarity": 0.0,
145
+ "eval_linker_valid": 0.9967245042492918,
146
+ "eval_loss": 0.357697993516922,
147
+ "eval_num_fragments": 3.0000885269121813,
148
+ "eval_poi_equal": 0.7862960339943342,
149
+ "eval_poi_graph_edit_distance": Infinity,
150
+ "eval_poi_graph_edit_distance_norm": Infinity,
151
+ "eval_poi_has_attachment_point(s)": 0.9579497167138811,
152
+ "eval_poi_heavy_atoms_difference": 1.0625885269121813,
153
+ "eval_poi_heavy_atoms_difference_norm": 0.030525122327743574,
154
+ "eval_poi_tanimoto_similarity": 0.0,
155
+ "eval_poi_valid": 0.9579497167138811,
156
+ "eval_reassembly": 0.5933958923512748,
157
+ "eval_reassembly_nostereo": 0.62296388101983,
158
+ "eval_runtime": 2314.6777,
159
+ "eval_samples_per_second": 4.88,
160
+ "eval_steps_per_second": 0.076,
161
+ "eval_tanimoto_similarity": 0.0,
162
+ "eval_valid": 0.9328080736543909,
163
+ "step": 80000
164
  }
165
  ],
166
  "logging_steps": 5000,
 
180
  "attributes": {}
181
  }
182
  },
183
+ "total_flos": 6.908210242100986e+17,
184
  "train_batch_size": 128,
185
  "trial_name": null,
186
  "trial_params": null