{
  "best_metric": 0.5623508616880247,
  "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_reduce-opt25/trial-number=3-learning_rate=1.1e-04-warmup_ratio=0.050-min_lr=0.000-factor=0.850/checkpoint-10000",
  "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.4528912305831909,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.7509,
      "step": 500
    },
    {
      "epoch": 0.09863878477017163,
      "grad_norm": 0.36168304085731506,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0993,
      "step": 1000
    },
    {
      "epoch": 0.14795817715525744,
      "grad_norm": 0.21691548824310303,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0324,
      "step": 1500
    },
    {
      "epoch": 0.19727756954034326,
      "grad_norm": 0.14681176841259003,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0176,
      "step": 2000
    },
    {
      "epoch": 0.24659696192542907,
      "grad_norm": 0.1438729166984558,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0119,
      "step": 2500
    },
    {
      "epoch": 0.2959163543105149,
      "grad_norm": 0.11427587270736694,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.009,
      "step": 3000
    },
    {
      "epoch": 0.3452357466956007,
      "grad_norm": 0.10182422399520874,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0074,
      "step": 3500
    },
    {
      "epoch": 0.3945551390806865,
      "grad_norm": 0.1254289746284485,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0082,
      "step": 4000
    },
    {
      "epoch": 0.4438745314657723,
      "grad_norm": 0.10901518911123276,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0062,
      "step": 4500
    },
    {
      "epoch": 0.49319392385085814,
      "grad_norm": 0.11559788137674332,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0065,
      "step": 5000
    },
    {
      "epoch": 0.49319392385085814,
      "eval_all_ligands_equal": 0.5236411842686699,
      "eval_e3_equal": 0.8001767565178965,
      "eval_e3_graph_edit_distance": 9.999999999999999e+63,
      "eval_e3_graph_edit_distance_norm": 1.0,
      "eval_e3_has_attachment_point(s)": 0.9847989394608926,
      "eval_e3_heavy_atoms_difference": 0.6638091029606716,
      "eval_e3_heavy_atoms_difference_norm": 0.014836414397080187,
      "eval_e3_tanimoto_similarity": 0.0,
      "eval_e3_valid": 0.9847989394608926,
      "eval_has_all_attachment_points": 0.9811754308440124,
      "eval_has_three_substructures": 0.9996464869642068,
      "eval_heavy_atoms_difference": 6.973221387538666,
      "eval_heavy_atoms_difference_norm": 0.09065782983430724,
      "eval_linker_equal": 0.7539549270879363,
      "eval_linker_graph_edit_distance": 9.999999999999999e+63,
      "eval_linker_graph_edit_distance_norm": 1.0,
      "eval_linker_has_attachment_point(s)": 0.9969951391957579,
      "eval_linker_heavy_atoms_difference": 0.559169244365886,
      "eval_linker_heavy_atoms_difference_norm": 0.012528181856226664,
      "eval_linker_tanimoto_similarity": 0.0,
      "eval_linker_valid": 0.9969951391957579,
      "eval_loss": 0.30445051193237305,
      "eval_num_fragments": 3.0001767565178965,
      "eval_poi_equal": 0.7479452054794521,
      "eval_poi_graph_edit_distance": 9.999999999999999e+63,
      "eval_poi_graph_edit_distance_norm": 1.0,
      "eval_poi_has_attachment_point(s)": 0.9374281926646045,
      "eval_poi_heavy_atoms_difference": 1.8859036676977463,
      "eval_poi_heavy_atoms_difference_norm": 0.05362304789611469,
      "eval_poi_tanimoto_similarity": 0.0,
      "eval_poi_valid": 0.9374281926646045,
      "eval_reassembly": 0.5313300927971719,
      "eval_reassembly_nostereo": 0.5539549270879364,
      "eval_runtime": 1550.2145,
      "eval_samples_per_second": 7.299,
      "eval_steps_per_second": 0.114,
      "eval_tanimoto_similarity": 0.0,
      "eval_valid": 0.921608484312859,
      "step": 5000
    },
    {
      "epoch": 0.542513316235944,
      "grad_norm": 0.11470098048448563,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0042,
      "step": 5500
    },
    {
      "epoch": 0.5918327086210298,
      "grad_norm": 0.11638347059488297,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0042,
      "step": 6000
    },
    {
      "epoch": 0.6411521010061156,
      "grad_norm": 0.07129918038845062,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0059,
      "step": 6500
    },
    {
      "epoch": 0.6904714933912014,
      "grad_norm": 0.044687844812870026,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0034,
      "step": 7000
    },
    {
      "epoch": 0.7397908857762873,
      "grad_norm": 0.0823349729180336,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0031,
      "step": 7500
    },
    {
      "epoch": 0.7397908857762873,
      "eval_all_ligands_equal": 0.5498895271763147,
      "eval_e3_equal": 0.8119310649580204,
      "eval_e3_graph_edit_distance": 9.999999999999999e+63,
      "eval_e3_graph_edit_distance_norm": 1.0,
      "eval_e3_has_attachment_point(s)": 0.9771100309323907,
      "eval_e3_heavy_atoms_difference": 0.8516129032258064,
      "eval_e3_heavy_atoms_difference_norm": 0.02148143499541932,
      "eval_e3_tanimoto_similarity": 0.0,
      "eval_e3_valid": 0.9771100309323907,
      "eval_has_all_attachment_points": 0.9779938135218736,
      "eval_has_three_substructures": 0.9984091913389306,
      "eval_heavy_atoms_difference": 7.066725585505965,
      "eval_heavy_atoms_difference_norm": 0.09269971150883742,
      "eval_linker_equal": 0.7787892178524083,
      "eval_linker_graph_edit_distance": 9.999999999999999e+63,
      "eval_linker_graph_edit_distance_norm": 1.0,
      "eval_linker_has_attachment_point(s)": 0.9923994697304463,
      "eval_linker_heavy_atoms_difference": 0.4481661511268228,
      "eval_linker_heavy_atoms_difference_norm": 0.006821181166453098,
      "eval_linker_tanimoto_similarity": 0.0,
      "eval_linker_valid": 0.9923994697304463,
      "eval_loss": 0.31453314423561096,
      "eval_num_fragments": 2.9998232434821035,
      "eval_poi_equal": 0.7662395050817499,
      "eval_poi_graph_edit_distance": 9.999999999999999e+63,
      "eval_poi_graph_edit_distance_norm": 1.0,
      "eval_poi_has_attachment_point(s)": 0.9374281926646045,
      "eval_poi_heavy_atoms_difference": 1.763057887759611,
      "eval_poi_heavy_atoms_difference_norm": 0.05254183317720355,
      "eval_poi_tanimoto_similarity": 0.0,
      "eval_poi_valid": 0.9374281926646045,
      "eval_reassembly": 0.5576668139637649,
      "eval_reassembly_nostereo": 0.5741051701281484,
      "eval_runtime": 1494.0608,
      "eval_samples_per_second": 7.573,
      "eval_steps_per_second": 0.118,
      "eval_tanimoto_similarity": 0.0,
      "eval_valid": 0.9133009279717189,
      "step": 7500
    },
    {
      "epoch": 0.789110278161373,
      "grad_norm": 0.0743485540151596,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0034,
      "step": 8000
    },
    {
      "epoch": 0.8384296705464589,
      "grad_norm": 0.08227942138910294,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0035,
      "step": 8500
    },
    {
      "epoch": 0.8877490629315447,
      "grad_norm": 0.07675650715827942,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.004,
      "step": 9000
    },
    {
      "epoch": 0.9370684553166305,
      "grad_norm": 0.060432229191064835,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.003,
      "step": 9500
    },
    {
      "epoch": 0.9863878477017163,
      "grad_norm": 0.06841175258159637,
      "learning_rate": 0.00010780369551982127,
      "loss": 0.0024,
      "step": 10000
    },
    {
      "epoch": 0.9863878477017163,
      "eval_all_ligands_equal": 0.5623508616880247,
      "eval_e3_equal": 0.8172337604949182,
      "eval_e3_graph_edit_distance": 9.999999999999999e+63,
      "eval_e3_graph_edit_distance_norm": 1.0,
      "eval_e3_has_attachment_point(s)": 0.9732213875386655,
      "eval_e3_heavy_atoms_difference": 0.8075121520106054,
      "eval_e3_heavy_atoms_difference_norm": 0.02178934615008687,
      "eval_e3_tanimoto_similarity": 0.0,
      "eval_e3_valid": 0.9732213875386655,
      "eval_has_all_attachment_points": 0.9811754308440124,
      "eval_has_three_substructures": 0.9995581087052585,
      "eval_heavy_atoms_difference": 7.874414494034467,
      "eval_heavy_atoms_difference_norm": 0.10424160867591652,
      "eval_linker_equal": 0.8018559434379142,
      "eval_linker_graph_edit_distance": 9.999999999999999e+63,
      "eval_linker_graph_edit_distance_norm": 1.0,
      "eval_linker_has_attachment_point(s)": 0.9961997348652232,
      "eval_linker_heavy_atoms_difference": 0.41741051701281484,
      "eval_linker_heavy_atoms_difference_norm": 0.008801670643119743,
      "eval_linker_tanimoto_similarity": 0.0,
      "eval_linker_valid": 0.9961997348652232,
      "eval_loss": 0.3179270923137665,
      "eval_num_fragments": 2.9999116217410515,
      "eval_poi_equal": 0.771895713654441,
      "eval_poi_graph_edit_distance": 9.999999999999999e+63,
      "eval_poi_graph_edit_distance_norm": 1.0,
      "eval_poi_has_attachment_point(s)": 0.9324790101634998,
      "eval_poi_heavy_atoms_difference": 1.9437030490499336,
      "eval_poi_heavy_atoms_difference_norm": 0.059822219414631145,
      "eval_poi_tanimoto_similarity": 0.0,
      "eval_poi_valid": 0.9324790101634998,
      "eval_reassembly": 0.5704816615112682,
      "eval_reassembly_nostereo": 0.5938135218736191,
      "eval_runtime": 1488.581,
      "eval_samples_per_second": 7.601,
      "eval_steps_per_second": 0.119,
      "eval_tanimoto_similarity": 0.0,
      "eval_valid": 0.9044631020768891,
      "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": "trial-number=3-learning_rate=1.1e-04-warmup_ratio=0.050-min_lr=0.000-factor=0.850",
  "trial_params": {
    "factor": 0.8500000000000001,
    "learning_rate": 0.00010780369551982127,
    "min_lr": 1.8229175490752733e-09,
    "warmup_ratio": 0.05
  }
}