PROTAC-Splitter-EncoderDecoder-lr_reduce-rand-smiles-train-0.25
/
last-checkpoint
/trainer_state.json
| { | |
| "best_metric": 0.5870219546742209, | |
| "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_reduce-rand-smiles-train-0.25/checkpoint-100000", | |
| "epoch": 39.447731755424066, | |
| "eval_steps": 20000, | |
| "global_step": 100000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 1.972386587771203, | |
| "grad_norm": 0.10182420909404755, | |
| "learning_rate": 5e-05, | |
| "loss": 0.1387, | |
| "step": 5000 | |
| }, | |
| { | |
| "epoch": 3.9447731755424065, | |
| "grad_norm": 0.08581843972206116, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0058, | |
| "step": 10000 | |
| }, | |
| { | |
| "epoch": 5.9171597633136095, | |
| "grad_norm": 0.05994575470685959, | |
| "learning_rate": 5e-05, | |
| "loss": 0.003, | |
| "step": 15000 | |
| }, | |
| { | |
| "epoch": 7.889546351084813, | |
| "grad_norm": 0.046327847987413406, | |
| "learning_rate": 5e-05, | |
| "loss": 0.002, | |
| "step": 20000 | |
| }, | |
| { | |
| "epoch": 9.861932938856016, | |
| "grad_norm": 0.050606127828359604, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0016, | |
| "step": 25000 | |
| }, | |
| { | |
| "epoch": 11.834319526627219, | |
| "grad_norm": 0.041334349662065506, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0012, | |
| "step": 30000 | |
| }, | |
| { | |
| "epoch": 13.806706114398422, | |
| "grad_norm": 0.04540463909506798, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0011, | |
| "step": 35000 | |
| }, | |
| { | |
| "epoch": 15.779092702169626, | |
| "grad_norm": 0.03727077320218086, | |
| "learning_rate": 5e-05, | |
| "loss": 0.001, | |
| "step": 40000 | |
| }, | |
| { | |
| "epoch": 17.75147928994083, | |
| "grad_norm": 0.02221524901688099, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0009, | |
| "step": 45000 | |
| }, | |
| { | |
| "epoch": 19.723865877712033, | |
| "grad_norm": 0.020608918741345406, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0008, | |
| "step": 50000 | |
| }, | |
| { | |
| "epoch": 21.696252465483234, | |
| "grad_norm": 0.018035519868135452, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0008, | |
| "step": 55000 | |
| }, | |
| { | |
| "epoch": 23.668639053254438, | |
| "grad_norm": 0.033636949956417084, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0007, | |
| "step": 60000 | |
| }, | |
| { | |
| "epoch": 25.641025641025642, | |
| "grad_norm": 0.051321662962436676, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0006, | |
| "step": 65000 | |
| }, | |
| { | |
| "epoch": 27.613412228796843, | |
| "grad_norm": 0.016928382217884064, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0006, | |
| "step": 70000 | |
| }, | |
| { | |
| "epoch": 29.585798816568047, | |
| "grad_norm": 0.02291979268193245, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0006, | |
| "step": 75000 | |
| }, | |
| { | |
| "epoch": 31.558185404339252, | |
| "grad_norm": 0.03006165847182274, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0006, | |
| "step": 80000 | |
| }, | |
| { | |
| "epoch": 31.558185404339252, | |
| "eval_all_ligands_equal": 0.5856055240793201, | |
| "eval_e3_equal": 0.8271954674220963, | |
| "eval_e3_graph_edit_distance": Infinity, | |
| "eval_e3_graph_edit_distance_norm": Infinity, | |
| "eval_e3_has_attachment_point(s)": 0.9770715297450425, | |
| "eval_e3_heavy_atoms_difference": 0.8041784702549575, | |
| "eval_e3_heavy_atoms_difference_norm": 0.02507045406737937, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 0.9770715297450425, | |
| "eval_has_all_attachment_points": 0.9890226628895185, | |
| "eval_has_three_substructures": 0.9999114730878187, | |
| "eval_heavy_atoms_difference": 5.752301699716714, | |
| "eval_heavy_atoms_difference_norm": 0.0757990141580953, | |
| "eval_linker_equal": 0.8324185552407932, | |
| "eval_linker_graph_edit_distance": Infinity, | |
| "eval_linker_graph_edit_distance_norm": Infinity, | |
| "eval_linker_has_attachment_point(s)": 0.9967245042492918, | |
| "eval_linker_heavy_atoms_difference": 0.2646954674220963, | |
| "eval_linker_heavy_atoms_difference_norm": 0.005140908527503901, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 0.9967245042492918, | |
| "eval_loss": 0.357697993516922, | |
| "eval_num_fragments": 3.0000885269121813, | |
| "eval_poi_equal": 0.7862960339943342, | |
| "eval_poi_graph_edit_distance": Infinity, | |
| "eval_poi_graph_edit_distance_norm": Infinity, | |
| "eval_poi_has_attachment_point(s)": 0.9579497167138811, | |
| "eval_poi_heavy_atoms_difference": 1.0625885269121813, | |
| "eval_poi_heavy_atoms_difference_norm": 0.030525122327743574, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 0.9579497167138811, | |
| "eval_reassembly": 0.5933958923512748, | |
| "eval_reassembly_nostereo": 0.62296388101983, | |
| "eval_runtime": 2314.6777, | |
| "eval_samples_per_second": 4.88, | |
| "eval_steps_per_second": 0.076, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 0.9328080736543909, | |
| "step": 80000 | |
| }, | |
| { | |
| "epoch": 33.53057199211045, | |
| "grad_norm": 0.015540308319032192, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0005, | |
| "step": 85000 | |
| }, | |
| { | |
| "epoch": 35.50295857988166, | |
| "grad_norm": 0.04315592721104622, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0005, | |
| "step": 90000 | |
| }, | |
| { | |
| "epoch": 37.47534516765286, | |
| "grad_norm": 0.017927605658769608, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0005, | |
| "step": 95000 | |
| }, | |
| { | |
| "epoch": 39.447731755424066, | |
| "grad_norm": 0.010922293178737164, | |
| "learning_rate": 5e-05, | |
| "loss": 0.0005, | |
| "step": 100000 | |
| }, | |
| { | |
| "epoch": 39.447731755424066, | |
| "eval_all_ligands_equal": 0.5870219546742209, | |
| "eval_e3_equal": 0.8276381019830028, | |
| "eval_e3_graph_edit_distance": Infinity, | |
| "eval_e3_graph_edit_distance_norm": Infinity, | |
| "eval_e3_has_attachment_point(s)": 0.9798158640226629, | |
| "eval_e3_heavy_atoms_difference": 0.4993803116147309, | |
| "eval_e3_heavy_atoms_difference_norm": 0.01531891728987192, | |
| "eval_e3_tanimoto_similarity": 0.0, | |
| "eval_e3_valid": 0.9798158640226629, | |
| "eval_has_all_attachment_points": 0.9888456090651558, | |
| "eval_has_three_substructures": 1.0, | |
| "eval_heavy_atoms_difference": 5.47078611898017, | |
| "eval_heavy_atoms_difference_norm": 0.07306279376337993, | |
| "eval_linker_equal": 0.8350743626062322, | |
| "eval_linker_graph_edit_distance": Infinity, | |
| "eval_linker_graph_edit_distance_norm": Infinity, | |
| "eval_linker_has_attachment_point(s)": 0.9969900849858357, | |
| "eval_linker_heavy_atoms_difference": 0.28001062322946174, | |
| "eval_linker_heavy_atoms_difference_norm": 0.0030083260769921197, | |
| "eval_linker_tanimoto_similarity": 0.0, | |
| "eval_linker_valid": 0.9969900849858357, | |
| "eval_loss": 0.36241990327835083, | |
| "eval_num_fragments": 3.0, | |
| "eval_poi_equal": 0.7936437677053825, | |
| "eval_poi_graph_edit_distance": Infinity, | |
| "eval_poi_graph_edit_distance_norm": Infinity, | |
| "eval_poi_has_attachment_point(s)": 0.9592776203966006, | |
| "eval_poi_heavy_atoms_difference": 1.1101274787535411, | |
| "eval_poi_heavy_atoms_difference_norm": 0.03114921880738521, | |
| "eval_poi_tanimoto_similarity": 0.0, | |
| "eval_poi_valid": 0.9592776203966006, | |
| "eval_reassembly": 0.5943696883852692, | |
| "eval_reassembly_nostereo": 0.6219900849858357, | |
| "eval_runtime": 2314.436, | |
| "eval_samples_per_second": 4.881, | |
| "eval_steps_per_second": 0.076, | |
| "eval_tanimoto_similarity": 0.0, | |
| "eval_valid": 0.9367032577903682, | |
| "step": 100000 | |
| } | |
| ], | |
| "logging_steps": 5000, | |
| "max_steps": 100000, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 40, | |
| "save_steps": 20000, | |
| "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.63511944471965e+17, | |
| "train_batch_size": 128, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |