Training in progress, step 80000, checkpoint

last-checkpoint/generation_config.json

@@ -1,5 +1,4 @@
 {
-  "_from_model_config": true,
   "bos_token_id": 0,
   "do_sample": true,
   "eos_token_id": 2,

last-checkpoint/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b9423ced92e1ead42dafd29660955040c4f9cbeb106ac16d725898dabb295a0b
+oid sha256:77158e1a93de65c4f3f2a040e94fe078b57c318e777bfd1a4df013ce130b5203
 size 409608164

last-checkpoint/optimizer.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7a8525f10d587fdf62ca0b5a216d729a480636349022a107c4c4f443ecd7e1a6
+oid sha256:e59c2bff8952fedefa351d5016f62be92e70cc9ae305cdfea00c51b8fe3997e2
 size 814647162

last-checkpoint/rng_state.pth

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ce9759301f561e437292986014727b714a5d6c291bf10ad6a3f2ba61221ee481
+oid sha256:e0ada99153844873ab029012909b759d30448c8245a7edf5c33f603b8234187b
 size 14244

last-checkpoint/scheduler.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ec08642f74697f111f64e755e2fc9bd211b07cc951ee155be7841c13ee3a6fae
+oid sha256:14015b6c173b9e7aa352e90e428e3d7c58f635929559378b29fbabecdeaa0149
 size 1192

last-checkpoint/trainer_state.json

@@ -1,9 +1,9 @@
 {
-  "best_metric": null,
-  "best_model_checkpoint": null,
-  "epoch": 5.918327086210298,
+  "best_metric": 0.5935729461756374,
+  "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_reduce-opt25/checkpoint-80000",
+  "epoch": 7.89110278161373,
   "eval_steps": 20000,
-  "global_step": 60000,
+  "global_step": 80000,
   "is_hyper_param_search": false,
   "is_local_process_zero": true,
   "is_world_process_zero": true,
@@ -91,6 +91,76 @@
       "learning_rate": 5e-05,
       "loss": 0.0006,
       "step": 60000
+    },
+    {
+      "epoch": 6.411521010061156,
+      "grad_norm": 0.01052697654813528,
+      "learning_rate": 5e-05,
+      "loss": 0.0006,
+      "step": 65000
+    },
+    {
+      "epoch": 6.904714933912015,
+      "grad_norm": 0.016045687720179558,
+      "learning_rate": 5e-05,
+      "loss": 0.0006,
+      "step": 70000
+    },
+    {
+      "epoch": 7.397908857762872,
+      "grad_norm": 0.025792840868234634,
+      "learning_rate": 5e-05,
+      "loss": 0.0005,
+      "step": 75000
+    },
+    {
+      "epoch": 7.89110278161373,
+      "grad_norm": 0.009077737107872963,
+      "learning_rate": 5e-05,
+      "loss": 0.0005,
+      "step": 80000
+    },
+    {
+      "epoch": 7.89110278161373,
+      "eval_all_ligands_equal": 0.5935729461756374,
+      "eval_e3_equal": 0.8268413597733711,
+      "eval_e3_graph_edit_distance": Infinity,
+      "eval_e3_graph_edit_distance_norm": Infinity,
+      "eval_e3_has_attachment_point(s)": 0.9716713881019831,
+      "eval_e3_heavy_atoms_difference": 0.90625,
+      "eval_e3_heavy_atoms_difference_norm": 0.033364542286953076,
+      "eval_e3_tanimoto_similarity": 0.0,
+      "eval_e3_valid": 0.9716713881019831,
+      "eval_has_all_attachment_points": 0.9788420679886686,
+      "eval_has_three_substructures": 0.9993803116147308,
+      "eval_heavy_atoms_difference": 9.06046388101983,
+      "eval_heavy_atoms_difference_norm": 0.12114217817938878,
+      "eval_linker_equal": 0.8393236543909348,
+      "eval_linker_graph_edit_distance": Infinity,
+      "eval_linker_graph_edit_distance_norm": Infinity,
+      "eval_linker_has_attachment_point(s)": 0.9961048158640227,
+      "eval_linker_heavy_atoms_difference": 0.22229107648725213,
+      "eval_linker_heavy_atoms_difference_norm": 0.0019467560966645548,
+      "eval_linker_tanimoto_similarity": 0.0,
+      "eval_linker_valid": 0.9961048158640227,
+      "eval_loss": 0.33752545714378357,
+      "eval_num_fragments": 2.999380311614731,
+      "eval_poi_equal": 0.7867386685552408,
+      "eval_poi_graph_edit_distance": 8.206444759206799e+62,
+      "eval_poi_graph_edit_distance_norm": Infinity,
+      "eval_poi_has_attachment_point(s)": 0.917935552407932,
+      "eval_poi_heavy_atoms_difference": 2.4516643059490084,
+      "eval_poi_heavy_atoms_difference_norm": 0.08128733092692968,
+      "eval_poi_tanimoto_similarity": 0.0,
+      "eval_poi_valid": 0.917935552407932,
+      "eval_reassembly": 0.6010977337110481,
+      "eval_reassembly_nostereo": 0.6211048158640227,
+      "eval_runtime": 2288.4581,
+      "eval_samples_per_second": 4.936,
+      "eval_steps_per_second": 0.077,
+      "eval_tanimoto_similarity": 0.0,
+      "eval_valid": 0.8892528328611898,
+      "step": 80000
     }
   ],
   "logging_steps": 5000,
@@ -110,7 +180,7 @@
       "attributes": {}
     }
   },
-  "total_flos": 5.1980516444269824e+17,
+  "total_flos": 6.930316474770202e+17,
   "train_batch_size": 128,
   "trial_name": null,
   "trial_params": null