{
  "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
}