| { |
| "best_metric": 0.0, |
| "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_reduce-opt25/trial-number=11-learning_rate=1.9e-04-warmup_ratio=0.050-min_lr=0.000-factor=0.830/checkpoint-5000", |
| "epoch": 0.49319392385085814, |
| "eval_steps": 2500, |
| "global_step": 5000, |
| "is_hyper_param_search": true, |
| "is_local_process_zero": true, |
| "is_world_process_zero": true, |
| "log_history": [ |
| { |
| "epoch": 0.049319392385085814, |
| "grad_norm": 0.3732318580150604, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 1.0002, |
| "step": 500 |
| }, |
| { |
| "epoch": 0.09863878477017163, |
| "grad_norm": 0.35028719902038574, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 0.2021, |
| "step": 1000 |
| }, |
| { |
| "epoch": 0.14795817715525744, |
| "grad_norm": 0.27330875396728516, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 0.1767, |
| "step": 1500 |
| }, |
| { |
| "epoch": 0.19727756954034326, |
| "grad_norm": 0.2673795223236084, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 0.1683, |
| "step": 2000 |
| }, |
| { |
| "epoch": 0.24659696192542907, |
| "grad_norm": 0.2766689956188202, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 0.1635, |
| "step": 2500 |
| }, |
| { |
| "epoch": 0.2959163543105149, |
| "grad_norm": 0.2843599319458008, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 0.1607, |
| "step": 3000 |
| }, |
| { |
| "epoch": 0.3452357466956007, |
| "grad_norm": 0.25838175415992737, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 0.1608, |
| "step": 3500 |
| }, |
| { |
| "epoch": 0.3945551390806865, |
| "grad_norm": 0.2386324256658554, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 0.1567, |
| "step": 4000 |
| }, |
| { |
| "epoch": 0.4438745314657723, |
| "grad_norm": 0.30100056529045105, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 0.1548, |
| "step": 4500 |
| }, |
| { |
| "epoch": 0.49319392385085814, |
| "grad_norm": 0.28358349204063416, |
| "learning_rate": 0.00018563759769650177, |
| "loss": 0.1545, |
| "step": 5000 |
| }, |
| { |
| "epoch": 0.49319392385085814, |
| "eval_all_ligands_equal": 0.0, |
| "eval_e3_equal": 0.011400795404330534, |
| "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.529297392841361, |
| "eval_e3_heavy_atoms_difference_norm": 0.23516541198896576, |
| "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.966151126822801, |
| "eval_heavy_atoms_difference_norm": 0.08358236778824747, |
| "eval_linker_equal": 0.0014140521431727795, |
| "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.30746796288113126, |
| "eval_linker_heavy_atoms_difference_norm": -0.2548477714690266, |
| "eval_linker_tanimoto_similarity": 0.0, |
| "eval_linker_valid": 1.0, |
| "eval_loss": 0.5128470063209534, |
| "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.7443216968625718, |
| "eval_poi_heavy_atoms_difference_norm": -0.11685097660916799, |
| "eval_poi_tanimoto_similarity": 0.0, |
| "eval_poi_valid": 1.0, |
| "eval_reassembly": 0.0, |
| "eval_reassembly_nostereo": 0.0, |
| "eval_runtime": 1180.6897, |
| "eval_samples_per_second": 9.583, |
| "eval_steps_per_second": 0.15, |
| "eval_tanimoto_similarity": 0.0, |
| "eval_valid": 1.0, |
| "step": 5000 |
| } |
| ], |
| "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": false |
| }, |
| "attributes": {} |
| } |
| }, |
| "total_flos": 4.33139477262336e+16, |
| "train_batch_size": 128, |
| "trial_name": "trial-number=11-learning_rate=1.9e-04-warmup_ratio=0.050-min_lr=0.000-factor=0.830", |
| "trial_params": { |
| "factor": 0.8300000000000001, |
| "learning_rate": 0.00018563759769650177, |
| "min_lr": 1.6647559250833267e-10, |
| "warmup_ratio": 0.05 |
| } |
| } |
|
|
