{
  "best_metric": 0.5992386685552408,
  "best_model_checkpoint": "/mimer/NOBACKUP/groups/naiss2023-6-290/stefano/models//PROTAC-Splitter-EncoderDecoder-lr_reduce-opt25/checkpoint-100000",
  "epoch": 9.863878477017163,
  "eval_steps": 20000,
  "global_step": 100000,
  "is_hyper_param_search": false,
  "is_local_process_zero": true,
  "is_world_process_zero": true,
  "log_history": [
    {
      "epoch": 0.49319392385085814,
      "grad_norm": 0.15789619088172913,
      "learning_rate": 5e-05,
      "loss": 0.1302,
      "step": 5000
    },
    {
      "epoch": 0.9863878477017163,
      "grad_norm": 0.08171849697828293,
      "learning_rate": 5e-05,
      "loss": 0.0042,
      "step": 10000
    },
    {
      "epoch": 1.4795817715525745,
      "grad_norm": 0.07135193794965744,
      "learning_rate": 5e-05,
      "loss": 0.0023,
      "step": 15000
    },
    {
      "epoch": 1.9727756954034326,
      "grad_norm": 0.03533194214105606,
      "learning_rate": 5e-05,
      "loss": 0.0017,
      "step": 20000
    },
    {
      "epoch": 2.465969619254291,
      "grad_norm": 0.03086615726351738,
      "learning_rate": 5e-05,
      "loss": 0.0012,
      "step": 25000
    },
    {
      "epoch": 2.959163543105149,
      "grad_norm": 0.04374952241778374,
      "learning_rate": 5e-05,
      "loss": 0.0011,
      "step": 30000
    },
    {
      "epoch": 3.4523574669560073,
      "grad_norm": 0.06215918809175491,
      "learning_rate": 5e-05,
      "loss": 0.0009,
      "step": 35000
    },
    {
      "epoch": 3.945551390806865,
      "grad_norm": 0.03970140591263771,
      "learning_rate": 5e-05,
      "loss": 0.0008,
      "step": 40000
    },
    {
      "epoch": 4.438745314657724,
      "grad_norm": 0.05877165496349335,
      "learning_rate": 5e-05,
      "loss": 0.0007,
      "step": 45000
    },
    {
      "epoch": 4.931939238508582,
      "grad_norm": 0.0312739722430706,
      "learning_rate": 5e-05,
      "loss": 0.0008,
      "step": 50000
    },
    {
      "epoch": 5.425133162359439,
      "grad_norm": 0.013028254732489586,
      "learning_rate": 5e-05,
      "loss": 0.0007,
      "step": 55000
    },
    {
      "epoch": 5.918327086210298,
      "grad_norm": 0.011430254206061363,
      "learning_rate": 5e-05,
      "loss": 0.0006,
      "step": 60000
    },
    {
      "epoch": 6.411521010061156,
      "grad_norm": 0.01052697654813528,
      "learning_rate": 5e-05,
      "loss": 0.0006,
      "step": 65000
    },
    {
      "epoch": 6.904714933912015,
      "grad_norm": 0.016045687720179558,
      "learning_rate": 5e-05,
      "loss": 0.0006,
      "step": 70000
    },
    {
      "epoch": 7.397908857762872,
      "grad_norm": 0.025792840868234634,
      "learning_rate": 5e-05,
      "loss": 0.0005,
      "step": 75000
    },
    {
      "epoch": 7.89110278161373,
      "grad_norm": 0.009077737107872963,
      "learning_rate": 5e-05,
      "loss": 0.0005,
      "step": 80000
    },
    {
      "epoch": 7.89110278161373,
      "eval_all_ligands_equal": 0.5935729461756374,
      "eval_e3_equal": 0.8268413597733711,
      "eval_e3_graph_edit_distance": Infinity,
      "eval_e3_graph_edit_distance_norm": Infinity,
      "eval_e3_has_attachment_point(s)": 0.9716713881019831,
      "eval_e3_heavy_atoms_difference": 0.90625,
      "eval_e3_heavy_atoms_difference_norm": 0.033364542286953076,
      "eval_e3_tanimoto_similarity": 0.0,
      "eval_e3_valid": 0.9716713881019831,
      "eval_has_all_attachment_points": 0.9788420679886686,
      "eval_has_three_substructures": 0.9993803116147308,
      "eval_heavy_atoms_difference": 9.06046388101983,
      "eval_heavy_atoms_difference_norm": 0.12114217817938878,
      "eval_linker_equal": 0.8393236543909348,
      "eval_linker_graph_edit_distance": Infinity,
      "eval_linker_graph_edit_distance_norm": Infinity,
      "eval_linker_has_attachment_point(s)": 0.9961048158640227,
      "eval_linker_heavy_atoms_difference": 0.22229107648725213,
      "eval_linker_heavy_atoms_difference_norm": 0.0019467560966645548,
      "eval_linker_tanimoto_similarity": 0.0,
      "eval_linker_valid": 0.9961048158640227,
      "eval_loss": 0.33752545714378357,
      "eval_num_fragments": 2.999380311614731,
      "eval_poi_equal": 0.7867386685552408,
      "eval_poi_graph_edit_distance": 8.206444759206799e+62,
      "eval_poi_graph_edit_distance_norm": Infinity,
      "eval_poi_has_attachment_point(s)": 0.917935552407932,
      "eval_poi_heavy_atoms_difference": 2.4516643059490084,
      "eval_poi_heavy_atoms_difference_norm": 0.08128733092692968,
      "eval_poi_tanimoto_similarity": 0.0,
      "eval_poi_valid": 0.917935552407932,
      "eval_reassembly": 0.6010977337110481,
      "eval_reassembly_nostereo": 0.6211048158640227,
      "eval_runtime": 2288.4581,
      "eval_samples_per_second": 4.936,
      "eval_steps_per_second": 0.077,
      "eval_tanimoto_similarity": 0.0,
      "eval_valid": 0.8892528328611898,
      "step": 80000
    },
    {
      "epoch": 8.384296705464589,
      "grad_norm": 0.010849849320948124,
      "learning_rate": 5e-05,
      "loss": 0.0005,
      "step": 85000
    },
    {
      "epoch": 8.877490629315448,
      "grad_norm": 0.011544693261384964,
      "learning_rate": 5e-05,
      "loss": 0.0005,
      "step": 90000
    },
    {
      "epoch": 9.370684553166305,
      "grad_norm": 0.009384240955114365,
      "learning_rate": 5e-05,
      "loss": 0.0005,
      "step": 95000
    },
    {
      "epoch": 9.863878477017163,
      "grad_norm": 0.008162745274603367,
      "learning_rate": 5e-05,
      "loss": 0.0005,
      "step": 100000
    },
    {
      "epoch": 9.863878477017163,
      "eval_all_ligands_equal": 0.5992386685552408,
      "eval_e3_equal": 0.8302053824362606,
      "eval_e3_graph_edit_distance": Infinity,
      "eval_e3_graph_edit_distance_norm": Infinity,
      "eval_e3_has_attachment_point(s)": 0.9731763456090652,
      "eval_e3_heavy_atoms_difference": 1.0454143059490084,
      "eval_e3_heavy_atoms_difference_norm": 0.03348467744282844,
      "eval_e3_tanimoto_similarity": 0.0,
      "eval_e3_valid": 0.9731763456090652,
      "eval_has_all_attachment_points": 0.9796388101983002,
      "eval_has_three_substructures": 0.9994688385269122,
      "eval_heavy_atoms_difference": 8.02443342776204,
      "eval_heavy_atoms_difference_norm": 0.10755971022893991,
      "eval_linker_equal": 0.8418909348441926,
      "eval_linker_graph_edit_distance": 3.718130311614731e+61,
      "eval_linker_graph_edit_distance_norm": Infinity,
      "eval_linker_has_attachment_point(s)": 0.9962818696883853,
      "eval_linker_heavy_atoms_difference": 0.16890934844192634,
      "eval_linker_heavy_atoms_difference_norm": -0.004560230649587742,
      "eval_linker_tanimoto_similarity": 0.0,
      "eval_linker_valid": 0.9962818696883853,
      "eval_loss": 0.34204375743865967,
      "eval_num_fragments": 3.0001770538243626,
      "eval_poi_equal": 0.7897485835694051,
      "eval_poi_graph_edit_distance": Infinity,
      "eval_poi_graph_edit_distance_norm": Infinity,
      "eval_poi_has_attachment_point(s)": 0.9305063739376771,
      "eval_poi_heavy_atoms_difference": 2.0482471671388103,
      "eval_poi_heavy_atoms_difference_norm": 0.06900840502631218,
      "eval_poi_tanimoto_similarity": 0.0,
      "eval_poi_valid": 0.9305063739376771,
      "eval_reassembly": 0.6072946175637394,
      "eval_reassembly_nostereo": 0.6260623229461756,
      "eval_runtime": 2311.618,
      "eval_samples_per_second": 4.887,
      "eval_steps_per_second": 0.077,
      "eval_tanimoto_similarity": 0.0,
      "eval_valid": 0.9027089235127479,
      "step": 100000
    }
  ],
  "logging_steps": 5000,
  "max_steps": 100000,
  "num_input_tokens_seen": 0,
  "num_train_epochs": 10,
  "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.663000870419031e+17,
  "train_batch_size": 128,
  "trial_name": null,
  "trial_params": null
}