| { | |
| "best_metric": 0.5462110481586402, | |
| "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_cosine_restarts-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.7575409412384033, | |
| "learning_rate": 4.465132736856968e-05, | |
| "loss": 1.4495, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.09863878477017163, | |
| "grad_norm": 0.6125224232673645, | |
| "learning_rate": 8.628481651367859e-06, | |
| "loss": 0.3014, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.14795817715525744, | |
| "grad_norm": 0.5499436259269714, | |
| "learning_rate": 4.3093350952626764e-05, | |
| "loss": 0.1515, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.19727756954034326, | |
| "grad_norm": 0.28163060545921326, | |
| "learning_rate": 6.906649047373284e-06, | |
| "loss": 0.0885, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.24659696192542907, | |
| "grad_norm": 0.26792627573013306, | |
| "learning_rate": 4.137151834863209e-05, | |
| "loss": 0.0576, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.2959163543105149, | |
| "grad_norm": 0.23422876000404358, | |
| "learning_rate": 5.348672631430346e-06, | |
| "loss": 0.0408, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.3452357466956007, | |
| "grad_norm": 0.25026798248291016, | |
| "learning_rate": 3.9501422739279976e-05, | |
| "loss": 0.0302, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.3945551390806865, | |
| "grad_norm": 0.18633298575878143, | |
| "learning_rate": 3.968661679220468e-06, | |
| "loss": 0.0237, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.4438745314657723, | |
| "grad_norm": 0.21757744252681732, | |
| "learning_rate": 3.750000000000008e-05, | |
| "loss": 0.0193, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 0.49319392385085814, | |
| "grad_norm": 0.14472977817058563, | |
| "learning_rate": 2.77911378362688e-06, | |
| "loss": 0.0156, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.49319392385085814, | |
| "eval_all_ligands_equal": 0.45963172804532576, | |
| "eval_e3_equal": 0.7815155807365439, | |
| "eval_e3_graph_edit_distance": Infinity, | |
| "eval_e3_graph_edit_distance_norm": Infinity, | |
| "eval_e3_has_attachment_point(s)": 0.9915899433427762, | |
| "eval_e3_heavy_atoms_difference": 0.39128895184135976, | |
| "eval_e3_heavy_atoms_difference_norm": 0.006079141519917903, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 0.9915899433427762, | |
| "eval_has_all_attachment_points": 0.986278328611898, | |
| "eval_has_three_substructures": 0.9988491501416431, | |
| "eval_heavy_atoms_difference": 7.0030099150141645, | |
| "eval_heavy_atoms_difference_norm": 0.09184213866701363, | |
| "eval_linker_equal": 0.6622698300283286, | |
| "eval_linker_graph_edit_distance": 3.5410764872521247e+61, | |
| "eval_linker_graph_edit_distance_norm": Infinity, | |
| "eval_linker_has_attachment_point(s)": 0.9964589235127479, | |
| "eval_linker_heavy_atoms_difference": 0.38048866855524077, | |
| "eval_linker_heavy_atoms_difference_norm": 0.0004655148396369131, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 0.9964589235127479, | |
| "eval_loss": 0.28911474347114563, | |
| "eval_num_fragments": 3.000619688385269, | |
| "eval_poi_equal": 0.7265403682719547, | |
| "eval_poi_graph_edit_distance": Infinity, | |
| "eval_poi_graph_edit_distance_norm": Infinity, | |
| "eval_poi_has_attachment_point(s)": 0.9251947592067988, | |
| "eval_poi_heavy_atoms_difference": 2.3129426345609065, | |
| "eval_poi_heavy_atoms_difference_norm": 0.07392839153429423, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 0.9251947592067988, | |
| "eval_reassembly": 0.4665368271954674, | |
| "eval_reassembly_nostereo": 0.4910587818696884, | |
| "eval_runtime": 2426.7651, | |
| "eval_samples_per_second": 4.655, | |
| "eval_steps_per_second": 0.073, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 0.9168732294617564, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.542513316235944, | |
| "grad_norm": 0.21385140717029572, | |
| "learning_rate": 3.5385375325047186e-05, | |
| "loss": 0.0135, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 0.5918327086210298, | |
| "grad_norm": 0.1257910281419754, | |
| "learning_rate": 1.7908016745981804e-06, | |
| "loss": 0.011, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.6411521010061156, | |
| "grad_norm": 0.11779133975505829, | |
| "learning_rate": 3.3176699082935504e-05, | |
| "loss": 0.0099, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 0.6904714933912014, | |
| "grad_norm": 0.13609211146831512, | |
| "learning_rate": 1.0126756596375464e-06, | |
| "loss": 0.0089, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.7397908857762873, | |
| "grad_norm": 0.13676239550113678, | |
| "learning_rate": 3.089397338773558e-05, | |
| "loss": 0.0077, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.7397908857762873, | |
| "eval_all_ligands_equal": 0.4898194050991501, | |
| "eval_e3_equal": 0.7908109065155807, | |
| "eval_e3_graph_edit_distance": Infinity, | |
| "eval_e3_graph_edit_distance_norm": Infinity, | |
| "eval_e3_has_attachment_point(s)": 0.9856586402266289, | |
| "eval_e3_heavy_atoms_difference": 0.48450779036827196, | |
| "eval_e3_heavy_atoms_difference_norm": 0.008794338048794908, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 0.9856586402266289, | |
| "eval_has_all_attachment_points": 0.9812322946175638, | |
| "eval_has_three_substructures": 0.9994688385269122, | |
| "eval_heavy_atoms_difference": 8.28390580736544, | |
| "eval_heavy_atoms_difference_norm": 0.10934640012787276, | |
| "eval_linker_equal": 0.7095432011331445, | |
| "eval_linker_graph_edit_distance": 5.6657223796034e+61, | |
| "eval_linker_graph_edit_distance_norm": Infinity, | |
| "eval_linker_has_attachment_point(s)": 0.9943342776203966, | |
| "eval_linker_heavy_atoms_difference": 0.448300283286119, | |
| "eval_linker_heavy_atoms_difference_norm": 0.008959724750780267, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 0.9943342776203966, | |
| "eval_loss": 0.30725741386413574, | |
| "eval_num_fragments": 3.0001770538243626, | |
| "eval_poi_equal": 0.736278328611898, | |
| "eval_poi_graph_edit_distance": Infinity, | |
| "eval_poi_graph_edit_distance_norm": Infinity, | |
| "eval_poi_has_attachment_point(s)": 0.9160764872521246, | |
| "eval_poi_heavy_atoms_difference": 2.5749822946175636, | |
| "eval_poi_heavy_atoms_difference_norm": 0.0832007262765964, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 0.9160764872521246, | |
| "eval_reassembly": 0.49752124645892354, | |
| "eval_reassembly_nostereo": 0.5220432011331445, | |
| "eval_runtime": 2398.3138, | |
| "eval_samples_per_second": 4.71, | |
| "eval_steps_per_second": 0.074, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 0.8987252124645893, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.789110278161373, | |
| "grad_norm": 0.12578321993350983, | |
| "learning_rate": 4.517825684323074e-07, | |
| "loss": 0.0066, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.8384296705464589, | |
| "grad_norm": 0.12456193566322327, | |
| "learning_rate": 2.8557870956831956e-05, | |
| "loss": 0.0065, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 0.8877490629315447, | |
| "grad_norm": 0.1042216420173645, | |
| "learning_rate": 1.1320193567289083e-07, | |
| "loss": 0.0054, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 0.9370684553166305, | |
| "grad_norm": 0.142706498503685, | |
| "learning_rate": 2.6189547895593596e-05, | |
| "loss": 0.0055, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 0.9863878477017163, | |
| "grad_norm": 0.10423340648412704, | |
| "learning_rate": 0.0, | |
| "loss": 0.0046, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 0.9863878477017163, | |
| "eval_all_ligands_equal": 0.5462110481586402, | |
| "eval_e3_equal": 0.8045325779036827, | |
| "eval_e3_graph_edit_distance": Infinity, | |
| "eval_e3_graph_edit_distance_norm": Infinity, | |
| "eval_e3_has_attachment_point(s)": 0.9896423512747875, | |
| "eval_e3_heavy_atoms_difference": 0.5553293201133145, | |
| "eval_e3_heavy_atoms_difference_norm": 0.013095491925927988, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 0.9896423512747875, | |
| "eval_has_all_attachment_points": 0.9836225212464589, | |
| "eval_has_three_substructures": 0.9991147308781869, | |
| "eval_heavy_atoms_difference": 7.0101805949008495, | |
| "eval_heavy_atoms_difference_norm": 0.09392218371981209, | |
| "eval_linker_equal": 0.7666430594900849, | |
| "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.31444759206798867, | |
| "eval_linker_heavy_atoms_difference_norm": 0.0032526462468232837, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 0.9961048158640227, | |
| "eval_loss": 0.3085635304450989, | |
| "eval_num_fragments": 2.9998229461756374, | |
| "eval_poi_equal": 0.7679709631728046, | |
| "eval_poi_graph_edit_distance": Infinity, | |
| "eval_poi_graph_edit_distance_norm": Infinity, | |
| "eval_poi_has_attachment_point(s)": 0.9272308781869688, | |
| "eval_poi_heavy_atoms_difference": 2.120839235127479, | |
| "eval_poi_heavy_atoms_difference_norm": 0.07192788253687527, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 0.9272308781869688, | |
| "eval_reassembly": 0.5543555240793201, | |
| "eval_reassembly_nostereo": 0.5795856940509915, | |
| "eval_runtime": 2370.5232, | |
| "eval_samples_per_second": 4.765, | |
| "eval_steps_per_second": 0.075, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 0.9157223796033994, | |
| "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 | |
| } | |