| { | |
| "best_metric": 0.0, | |
| "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_reduce-opt25-rand-smiles/trial-number=19-learning_rate=1.8e-04-warmup_ratio=0.100-min_lr=0.000-factor=0.930/checkpoint-5000", | |
| "epoch": 0.9863878477017163, | |
| "eval_steps": 2500, | |
| "global_step": 10000, | |
| "is_hyper_param_search": true, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.049319392385085814, | |
| "grad_norm": 0.4460906684398651, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 1.0142, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.09863878477017163, | |
| "grad_norm": 0.3451838791370392, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.2001, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.14795817715525744, | |
| "grad_norm": 0.24968214333057404, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1756, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.19727756954034326, | |
| "grad_norm": 0.2555454969406128, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.168, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.24659696192542907, | |
| "grad_norm": 0.28854918479919434, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1633, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.2959163543105149, | |
| "grad_norm": 0.2615298628807068, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1605, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 0.3452357466956007, | |
| "grad_norm": 0.25099700689315796, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1589, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 0.3945551390806865, | |
| "grad_norm": 0.2410382330417633, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1569, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 0.4438745314657723, | |
| "grad_norm": 0.2645629644393921, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1554, | |
| "step": 4500 | |
| }, | |
| { | |
| "epoch": 0.49319392385085814, | |
| "grad_norm": 0.2651103734970093, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1543, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.49319392385085814, | |
| "eval_all_ligands_equal": 0.0, | |
| "eval_e3_equal": 0.01449403446752099, | |
| "eval_e3_graph_edit_distance": 9.999999999999999e+63, | |
| "eval_e3_graph_edit_distance_norm": 1.0, | |
| "eval_e3_has_attachment_point(s)": 1.0, | |
| "eval_e3_heavy_atoms_difference": 8.672381794078657, | |
| "eval_e3_heavy_atoms_difference_norm": 0.2408192904697644, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 1.0, | |
| "eval_has_all_attachment_points": 1.0, | |
| "eval_has_three_substructures": 1.0, | |
| "eval_heavy_atoms_difference": 8.089173663278833, | |
| "eval_heavy_atoms_difference_norm": 0.07232688019696262, | |
| "eval_linker_equal": 0.0016791869200176756, | |
| "eval_linker_graph_edit_distance": 9.999999999999999e+63, | |
| "eval_linker_graph_edit_distance_norm": 1.0, | |
| "eval_linker_has_attachment_point(s)": 1.0, | |
| "eval_linker_heavy_atoms_difference": -1.4842244807777287, | |
| "eval_linker_heavy_atoms_difference_norm": -0.34404268961334245, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 1.0, | |
| "eval_loss": 0.5216243863105774, | |
| "eval_num_fragments": 3.0, | |
| "eval_poi_equal": 0.0016791869200176756, | |
| "eval_poi_graph_edit_distance": 9.999999999999999e+63, | |
| "eval_poi_graph_edit_distance_norm": 1.0, | |
| "eval_poi_has_attachment_point(s)": 1.0, | |
| "eval_poi_heavy_atoms_difference": 0.9010163499779055, | |
| "eval_poi_heavy_atoms_difference_norm": -0.11167547976496513, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 1.0, | |
| "eval_reassembly": 0.0, | |
| "eval_reassembly_nostereo": 0.0, | |
| "eval_runtime": 1193.9687, | |
| "eval_samples_per_second": 9.477, | |
| "eval_steps_per_second": 0.148, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 1.0, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 0.542513316235944, | |
| "grad_norm": 0.27524861693382263, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1543, | |
| "step": 5500 | |
| }, | |
| { | |
| "epoch": 0.5918327086210298, | |
| "grad_norm": 0.2908000648021698, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1523, | |
| "step": 6000 | |
| }, | |
| { | |
| "epoch": 0.6411521010061156, | |
| "grad_norm": 0.27331531047821045, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1527, | |
| "step": 6500 | |
| }, | |
| { | |
| "epoch": 0.6904714933912014, | |
| "grad_norm": 0.3615107834339142, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1514, | |
| "step": 7000 | |
| }, | |
| { | |
| "epoch": 0.7397908857762873, | |
| "grad_norm": 0.29473376274108887, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1503, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.7397908857762873, | |
| "eval_all_ligands_equal": 0.0, | |
| "eval_e3_equal": 0.006716747680070702, | |
| "eval_e3_graph_edit_distance": 9.999999999999999e+63, | |
| "eval_e3_graph_edit_distance_norm": 1.0, | |
| "eval_e3_has_attachment_point(s)": 1.0, | |
| "eval_e3_heavy_atoms_difference": -6.550419796730004, | |
| "eval_e3_heavy_atoms_difference_norm": -0.3544036689079038, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 1.0, | |
| "eval_has_all_attachment_points": 1.0, | |
| "eval_has_three_substructures": 1.0, | |
| "eval_heavy_atoms_difference": -20.141140079540435, | |
| "eval_heavy_atoms_difference_norm": -0.3160320293892404, | |
| "eval_linker_equal": 0.001590808661069377, | |
| "eval_linker_graph_edit_distance": 9.999999999999999e+63, | |
| "eval_linker_graph_edit_distance_norm": 1.0, | |
| "eval_linker_has_attachment_point(s)": 1.0, | |
| "eval_linker_heavy_atoms_difference": -2.135218736190897, | |
| "eval_linker_heavy_atoms_difference_norm": -0.3955390343159519, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 1.0, | |
| "eval_loss": 0.5303946137428284, | |
| "eval_num_fragments": 3.0, | |
| "eval_poi_equal": 0.001237295625276182, | |
| "eval_poi_graph_edit_distance": 9.999999999999999e+63, | |
| "eval_poi_graph_edit_distance_norm": 1.0, | |
| "eval_poi_has_attachment_point(s)": 1.0, | |
| "eval_poi_heavy_atoms_difference": -11.455501546619532, | |
| "eval_poi_heavy_atoms_difference_norm": -0.565083423857184, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 1.0, | |
| "eval_reassembly": 0.0, | |
| "eval_reassembly_nostereo": 0.0, | |
| "eval_runtime": 1564.7984, | |
| "eval_samples_per_second": 7.231, | |
| "eval_steps_per_second": 0.113, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 1.0, | |
| "step": 7500 | |
| }, | |
| { | |
| "epoch": 0.789110278161373, | |
| "grad_norm": 0.28325650095939636, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1495, | |
| "step": 8000 | |
| }, | |
| { | |
| "epoch": 0.8384296705464589, | |
| "grad_norm": 0.3338431119918823, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1498, | |
| "step": 8500 | |
| }, | |
| { | |
| "epoch": 0.8877490629315447, | |
| "grad_norm": 0.2506033182144165, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1531, | |
| "step": 9000 | |
| }, | |
| { | |
| "epoch": 0.9370684553166305, | |
| "grad_norm": 0.3380557894706726, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1481, | |
| "step": 9500 | |
| }, | |
| { | |
| "epoch": 0.9863878477017163, | |
| "grad_norm": 0.3548344075679779, | |
| "learning_rate": 0.0001809696468670689, | |
| "loss": 0.1476, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 0.9863878477017163, | |
| "eval_all_ligands_equal": 0.0, | |
| "eval_e3_equal": 0.005479452054794521, | |
| "eval_e3_graph_edit_distance": 9.999999999999999e+63, | |
| "eval_e3_graph_edit_distance_norm": 1.0, | |
| "eval_e3_has_attachment_point(s)": 1.0, | |
| "eval_e3_heavy_atoms_difference": -5.9238179407865665, | |
| "eval_e3_heavy_atoms_difference_norm": -0.33000060419236454, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 1.0, | |
| "eval_has_all_attachment_points": 1.0, | |
| "eval_has_three_substructures": 1.0, | |
| "eval_heavy_atoms_difference": -9.478391515687141, | |
| "eval_heavy_atoms_difference_norm": -0.16956106665343912, | |
| "eval_linker_equal": 0.0011489173663278833, | |
| "eval_linker_graph_edit_distance": 9.999999999999999e+63, | |
| "eval_linker_graph_edit_distance_norm": 1.0, | |
| "eval_linker_has_attachment_point(s)": 1.0, | |
| "eval_linker_heavy_atoms_difference": 0.8265134776844896, | |
| "eval_linker_heavy_atoms_difference_norm": -0.16816308209299535, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 1.0, | |
| "eval_loss": 0.5292195081710815, | |
| "eval_num_fragments": 3.0, | |
| "eval_poi_equal": 0.0016791869200176756, | |
| "eval_poi_graph_edit_distance": 9.999999999999999e+63, | |
| "eval_poi_graph_edit_distance_norm": 1.0, | |
| "eval_poi_has_attachment_point(s)": 1.0, | |
| "eval_poi_heavy_atoms_difference": -4.381087052585064, | |
| "eval_poi_heavy_atoms_difference_norm": -0.3046896586238993, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 1.0, | |
| "eval_reassembly": 0.0, | |
| "eval_reassembly_nostereo": 0.0, | |
| "eval_runtime": 1608.355, | |
| "eval_samples_per_second": 7.035, | |
| "eval_steps_per_second": 0.11, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 1.0, | |
| "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.729673191023411e+16, | |
| "train_batch_size": 128, | |
| "trial_name": "trial-number=19-learning_rate=1.8e-04-warmup_ratio=0.100-min_lr=0.000-factor=0.930", | |
| "trial_params": { | |
| "factor": 0.93, | |
| "learning_rate": 0.0001809696468670689, | |
| "min_lr": 6.698351404357494e-10, | |
| "warmup_ratio": 0.09999999999999999 | |
| } | |
| } | |