ribesstefano commited on
Commit
2c50235
·
verified ·
1 Parent(s): 510285b

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:ee12323d3de40cc6fec2e53ef895e1b7ced882c5f1e301da444d2d3a86df81e2
3
  size 409608164
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2790709e38d495ed69451e2e98588967f60dac42ae76070227ad36e62492bafe
3
  size 409608164
last-checkpoint/optimizer.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:14679abceec2775a9a53dc56e0f88e04155e7d5239080dd3f62cd447309b6fcf
3
  size 814647162
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0ea8c05c52169b3d03aae647dfe51ba544b26ad04087e18fefc7c8ea6fd417c
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:1d820f2235ccb34f6ad754b4400efd96293f074f257b91237679dc0697b87a58
3
  size 14244
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4093fe9aef15ebbc7ba8817fd7aceacf673b76c0d65d6c1fb129733fa8a3765f
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:3b1f708867db1a95ab2c4f1689ddb7f69e9949ed92954bdde4a7404728d9757d
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": 11.836654172420596,
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.0008,
93
  "step": 60000
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
94
  }
95
  ],
96
  "logging_steps": 5000,
@@ -110,7 +180,7 @@
110
  "attributes": {}
111
  }
112
  },
113
- "total_flos": 5.2004587461159706e+17,
114
  "train_batch_size": 128,
115
  "trial_name": null,
116
  "trial_params": null
 
1
  {
2
+ "best_metric": 0.5903859773371105,
3
+ "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_reduce-rand-smiles-train-0.5/checkpoint-80000",
4
+ "epoch": 15.78220556322746,
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.0008,
93
  "step": 60000
94
+ },
95
+ {
96
+ "epoch": 12.823042020122312,
97
+ "grad_norm": 0.027352627366781235,
98
+ "learning_rate": 5e-05,
99
+ "loss": 0.0009,
100
+ "step": 65000
101
+ },
102
+ {
103
+ "epoch": 13.80942986782403,
104
+ "grad_norm": 0.013403774239122868,
105
+ "learning_rate": 5e-05,
106
+ "loss": 0.0006,
107
+ "step": 70000
108
+ },
109
+ {
110
+ "epoch": 14.795817715525745,
111
+ "grad_norm": 0.024653641507029533,
112
+ "learning_rate": 5e-05,
113
+ "loss": 0.0006,
114
+ "step": 75000
115
+ },
116
+ {
117
+ "epoch": 15.78220556322746,
118
+ "grad_norm": 0.01882922276854515,
119
+ "learning_rate": 5e-05,
120
+ "loss": 0.0006,
121
+ "step": 80000
122
+ },
123
+ {
124
+ "epoch": 15.78220556322746,
125
+ "eval_all_ligands_equal": 0.5903859773371105,
126
+ "eval_e3_equal": 0.8249822946175638,
127
+ "eval_e3_graph_edit_distance": Infinity,
128
+ "eval_e3_graph_edit_distance_norm": Infinity,
129
+ "eval_e3_has_attachment_point(s)": 0.9796388101983002,
130
+ "eval_e3_heavy_atoms_difference": 0.7439801699716714,
131
+ "eval_e3_heavy_atoms_difference_norm": 0.02207718155625103,
132
+ "eval_e3_tanimoto_similarity": 0.0,
133
+ "eval_e3_valid": 0.9796388101983002,
134
+ "eval_has_all_attachment_points": 0.9878718130311614,
135
+ "eval_has_three_substructures": 0.9993803116147308,
136
+ "eval_heavy_atoms_difference": 5.907046742209632,
137
+ "eval_heavy_atoms_difference_norm": 0.07815233287632471,
138
+ "eval_linker_equal": 0.8371104815864022,
139
+ "eval_linker_graph_edit_distance": Infinity,
140
+ "eval_linker_graph_edit_distance_norm": Infinity,
141
+ "eval_linker_has_attachment_point(s)": 0.9952195467422096,
142
+ "eval_linker_heavy_atoms_difference": 0.1988314447592068,
143
+ "eval_linker_heavy_atoms_difference_norm": -0.001430759237988029,
144
+ "eval_linker_tanimoto_similarity": 0.0,
145
+ "eval_linker_valid": 0.9952195467422096,
146
+ "eval_loss": 0.3654560148715973,
147
+ "eval_num_fragments": 2.9995573654390935,
148
+ "eval_poi_equal": 0.785056657223796,
149
+ "eval_poi_graph_edit_distance": Infinity,
150
+ "eval_poi_graph_edit_distance_norm": Infinity,
151
+ "eval_poi_has_attachment_point(s)": 0.9528151558073654,
152
+ "eval_poi_heavy_atoms_difference": 1.4796388101983002,
153
+ "eval_poi_heavy_atoms_difference_norm": 0.04765018378193289,
154
+ "eval_poi_tanimoto_similarity": 0.0,
155
+ "eval_poi_valid": 0.9528151558073654,
156
+ "eval_reassembly": 0.5977337110481586,
157
+ "eval_reassembly_nostereo": 0.6278328611898017,
158
+ "eval_runtime": 2302.0693,
159
+ "eval_samples_per_second": 4.907,
160
+ "eval_steps_per_second": 0.077,
161
+ "eval_tanimoto_similarity": 0.0,
162
+ "eval_valid": 0.9314801699716714,
163
+ "step": 80000
164
  }
165
  ],
166
  "logging_steps": 5000,
 
180
  "attributes": {}
181
  }
182
  },
183
+ "total_flos": 6.933784829879263e+17,
184
  "train_batch_size": 128,
185
  "trial_name": null,
186
  "trial_params": null