| { | |
| "best_metric": 0.5477160056657224, | |
| "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_cosine-opt25/checkpoint-10000", | |
| "epoch": 0.9863878477017163, | |
| "eval_steps": 2500, | |
| "global_step": 10000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.049319392385085814, | |
| "grad_norm": 0.9255660772323608, | |
| "learning_rate": 3.57653791130186e-05, | |
| "loss": 2.3662, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.09863878477017163, | |
| "grad_norm": 0.6119515895843506, | |
| "learning_rate": 4.987090502485352e-05, | |
| "loss": 0.3224, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.14795817715525744, | |
| "grad_norm": 0.4047534763813019, | |
| "learning_rate": 4.9090579537757233e-05, | |
| "loss": 0.1093, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.19727756954034326, | |
| "grad_norm": 0.3516473174095154, | |
| "learning_rate": 4.762477336752997e-05, | |
| "loss": 0.0539, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.24659696192542907, | |
| "grad_norm": 0.2518511414527893, | |
| "learning_rate": 4.5515195009118114e-05, | |
| "loss": 0.0324, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.2959163543105149, | |
| "grad_norm": 0.18100735545158386, | |
| "learning_rate": 4.282187104726926e-05, | |
| "loss": 0.0224, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.3452357466956007, | |
| "grad_norm": 0.17784442007541656, | |
| "learning_rate": 3.962143814181834e-05, | |
| "loss": 0.0167, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.3945551390806865, | |
| "grad_norm": 0.15665775537490845, | |
| "learning_rate": 3.600496238484107e-05, | |
| "loss": 0.0127, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.4438745314657723, | |
| "grad_norm": 0.190831258893013, | |
| "learning_rate": 3.207534807835037e-05, | |
| "loss": 0.0101, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 0.49319392385085814, | |
| "grad_norm": 0.17124462127685547, | |
| "learning_rate": 2.7944409663963666e-05, | |
| "loss": 0.0086, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.49319392385085814, | |
| "eval_all_ligands_equal": 0.4899079320113314, | |
| "eval_e3_equal": 0.7831975920679887, | |
| "eval_e3_graph_edit_distance": Infinity, | |
| "eval_e3_graph_edit_distance_norm": Infinity, | |
| "eval_e3_has_attachment_point(s)": 0.9948654390934845, | |
| "eval_e3_heavy_atoms_difference": 0.35915368271954673, | |
| "eval_e3_heavy_atoms_difference_norm": 0.004063330129381431, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 0.9948654390934845, | |
| "eval_has_all_attachment_points": 0.9820290368271954, | |
| "eval_has_three_substructures": 0.9988491501416431, | |
| "eval_heavy_atoms_difference": 7.435021246458923, | |
| "eval_heavy_atoms_difference_norm": 0.09742815646256632, | |
| "eval_linker_equal": 0.7091890934844193, | |
| "eval_linker_graph_edit_distance": 4.603399433427762e+61, | |
| "eval_linker_graph_edit_distance_norm": 0.054428620142835865, | |
| "eval_linker_has_attachment_point(s)": 0.9953966005665722, | |
| "eval_linker_heavy_atoms_difference": 0.6527089235127479, | |
| "eval_linker_heavy_atoms_difference_norm": 0.02000991525067224, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 0.9953966005665722, | |
| "eval_loss": 0.293075829744339, | |
| "eval_num_fragments": 3.0004426345609065, | |
| "eval_poi_equal": 0.7415899433427762, | |
| "eval_poi_graph_edit_distance": Infinity, | |
| "eval_poi_graph_edit_distance_norm": Infinity, | |
| "eval_poi_has_attachment_point(s)": 0.9247521246458924, | |
| "eval_poi_heavy_atoms_difference": 2.4198831444759206, | |
| "eval_poi_heavy_atoms_difference_norm": 0.07853231753644796, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 0.9247521246458924, | |
| "eval_reassembly": 0.4971671388101983, | |
| "eval_reassembly_nostereo": 0.5238137393767706, | |
| "eval_runtime": 2417.0464, | |
| "eval_samples_per_second": 4.673, | |
| "eval_steps_per_second": 0.073, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 0.9182011331444759, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.542513316235944, | |
| "grad_norm": 0.12277983129024506, | |
| "learning_rate": 2.372969012074575e-05, | |
| "loss": 0.007, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 0.5918327086210298, | |
| "grad_norm": 0.11204977333545685, | |
| "learning_rate": 1.9551116361503457e-05, | |
| "loss": 0.006, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.6411521010061156, | |
| "grad_norm": 0.1196436733007431, | |
| "learning_rate": 1.55275867959039e-05, | |
| "loss": 0.0051, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 0.6904714933912014, | |
| "grad_norm": 0.12161414325237274, | |
| "learning_rate": 1.1773588158934163e-05, | |
| "loss": 0.0045, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.7397908857762873, | |
| "grad_norm": 0.14829416573047638, | |
| "learning_rate": 8.395937870495241e-06, | |
| "loss": 0.004, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.7397908857762873, | |
| "eval_all_ligands_equal": 0.5380665722379604, | |
| "eval_e3_equal": 0.8043555240793201, | |
| "eval_e3_graph_edit_distance": Infinity, | |
| "eval_e3_graph_edit_distance_norm": Infinity, | |
| "eval_e3_has_attachment_point(s)": 0.9923866855524079, | |
| "eval_e3_heavy_atoms_difference": 0.4004957507082153, | |
| "eval_e3_heavy_atoms_difference_norm": 0.00806031722408992, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 0.9923866855524079, | |
| "eval_has_all_attachment_points": 0.9817634560906515, | |
| "eval_has_three_substructures": 0.9977868271954674, | |
| "eval_heavy_atoms_difference": 6.986809490084986, | |
| "eval_heavy_atoms_difference_norm": 0.09237955806043094, | |
| "eval_linker_equal": 0.7604461756373938, | |
| "eval_linker_graph_edit_distance": 5.93130311614731e+61, | |
| "eval_linker_graph_edit_distance_norm": Infinity, | |
| "eval_linker_has_attachment_point(s)": 0.9940686968838527, | |
| "eval_linker_heavy_atoms_difference": 0.24982294617563738, | |
| "eval_linker_heavy_atoms_difference_norm": -0.0010343910857730165, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 0.9940686968838527, | |
| "eval_loss": 0.30975914001464844, | |
| "eval_num_fragments": 3.000619688385269, | |
| "eval_poi_equal": 0.7602691218130312, | |
| "eval_poi_graph_edit_distance": Infinity, | |
| "eval_poi_graph_edit_distance_norm": Infinity, | |
| "eval_poi_has_attachment_point(s)": 0.9225389518413598, | |
| "eval_poi_heavy_atoms_difference": 2.202372521246459, | |
| "eval_poi_heavy_atoms_difference_norm": 0.07091974582407527, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 0.9225389518413598, | |
| "eval_reassembly": 0.5450601983002833, | |
| "eval_reassembly_nostereo": 0.5729461756373938, | |
| "eval_runtime": 2364.3909, | |
| "eval_samples_per_second": 4.778, | |
| "eval_steps_per_second": 0.075, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 0.9147485835694051, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.789110278161373, | |
| "grad_norm": 0.11315891146659851, | |
| "learning_rate": 5.490744620021837e-06, | |
| "loss": 0.0036, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.8384296705464589, | |
| "grad_norm": 0.10538846999406815, | |
| "learning_rate": 3.1406736605785858e-06, | |
| "loss": 0.0034, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 0.8877490629315447, | |
| "grad_norm": 0.09490078687667847, | |
| "learning_rate": 1.412594626586694e-06, | |
| "loss": 0.0032, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 0.9370684553166305, | |
| "grad_norm": 0.08204534649848938, | |
| "learning_rate": 3.5567880489165705e-07, | |
| "loss": 0.0031, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 0.9863878477017163, | |
| "grad_norm": 0.10336631536483765, | |
| "learning_rate": 0.0, | |
| "loss": 0.003, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 0.9863878477017163, | |
| "eval_all_ligands_equal": 0.5477160056657224, | |
| "eval_e3_equal": 0.8034702549575071, | |
| "eval_e3_graph_edit_distance": Infinity, | |
| "eval_e3_graph_edit_distance_norm": Infinity, | |
| "eval_e3_has_attachment_point(s)": 0.9942457507082153, | |
| "eval_e3_heavy_atoms_difference": 0.3627832861189802, | |
| "eval_e3_heavy_atoms_difference_norm": 0.004441809193160064, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 0.9942457507082153, | |
| "eval_has_all_attachment_points": 0.9856586402266289, | |
| "eval_has_three_substructures": 0.9983179886685553, | |
| "eval_heavy_atoms_difference": 6.492917847025495, | |
| "eval_heavy_atoms_difference_norm": 0.08538313201004577, | |
| "eval_linker_equal": 0.7725743626062322, | |
| "eval_linker_graph_edit_distance": Infinity, | |
| "eval_linker_graph_edit_distance_norm": Infinity, | |
| "eval_linker_has_attachment_point(s)": 0.9951310198300283, | |
| "eval_linker_heavy_atoms_difference": 0.32524787535410765, | |
| "eval_linker_heavy_atoms_difference_norm": 0.005000241007082242, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 0.9951310198300283, | |
| "eval_loss": 0.312394380569458, | |
| "eval_num_fragments": 3.0007967422096318, | |
| "eval_poi_equal": 0.7672627478753541, | |
| "eval_poi_graph_edit_distance": Infinity, | |
| "eval_poi_graph_edit_distance_norm": Infinity, | |
| "eval_poi_has_attachment_point(s)": 0.9293555240793201, | |
| "eval_poi_heavy_atoms_difference": 2.0848973087818696, | |
| "eval_poi_heavy_atoms_difference_norm": 0.06681110633224353, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 0.9293555240793201, | |
| "eval_reassembly": 0.5548866855524079, | |
| "eval_reassembly_nostereo": 0.5845432011331445, | |
| "eval_runtime": 2338.9582, | |
| "eval_samples_per_second": 4.83, | |
| "eval_steps_per_second": 0.076, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 0.9231586402266289, | |
| "step": 10000 | |
| } | |
| ], | |
| "logging_steps": 500, | |
| "max_steps": 10000, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 1, | |
| "save_steps": 5000, | |
| "stateful_callbacks": { | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 8.663036348367667e+16, | |
| "train_batch_size": 128, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |