{
  "train_losses": [
    2136.5024682617186,
    1198.3308361816407,
    1161.5707946777343,
    1123.9894409179688,
    1090.3879553222657,
    1057.5052581787108,
    1034.7834246826171,
    1018.692387084961,
    1008.2613201904297,
    1000.2829901123047,
    993.2367407226562,
    987.9076995849609,
    984.4344122314453,
    982.5031854248047,
    980.7457598876953
  ],
  "test_losses": [
    1273.5531799316407,
    1226.4591247558594,
    1197.41689453125,
    1164.7423767089845,
    1098.9119689941406,
    1046.8178466796876,
    1028.4834930419922,
    1017.2713775634766,
    1018.4531524658203,
    1009.554721069336,
    1002.5865447998046,
    985.553515625,
    986.9031860351563,
    991.0865600585937,
    990.1863067626953
  ],
  "config": {
    "epochs": 15,
    "batch_size": 32,
    "max_learning_rate": 0.001,
    "sequence_size": 32,
    "shuffle_batches": true,
    "shuffle_within_batch": true,
    "vae_config": {
      "latent_dim": 96,
      "encoder_dropout": 0.1,
      "decoder_dropout": 0.1,
      "initial_mi_beta": 1.0,
      "final_mi_beta": 1.0,
      "mi_beta_shape": "constant",
      "initial_tc_beta": 0.0,
      "final_tc_beta": 10.0,
      "tc_beta_shape": "custom",
      "initial_dw_beta": 0.2,
      "final_dw_beta": 1.0,
      "dw_beta_shape": "custom",
      "warmup_epoch_ratio": 0.2,
      "free_bits": 0.75,
      "focal_loss_alpha": 0.5,
      "focal_loss_gamma": 2.0
    },
    "adaptive_weighting": {
      "initial_mi_beta": 1.0,
      "final_mi_beta": 1.0,
      "mi_beta_shape": "constant",
      "initial_tc_beta": 0.0,
      "final_tc_beta": 10.0,
      "tc_beta_shape": "custom",
      "initial_dw_beta": 0.2,
      "final_dw_beta": 1.0,
      "dw_beta_shape": "custom",
      "warmup_epoch_ratio": 0.2
    },
    "regularization": {
      "encoder_dropout": 0.1,
      "decoder_dropout": 0.1,
      "free_bits": 0.75,
      "focal_loss_alpha": 0.5,
      "focal_loss_gamma": 2.0
    },
    "early_stopping": {
      "enabled": false,
      "patience": 3,
      "min_delta": 0.01,
      "triggered": false,
      "best_epoch": null
    }
  },
  "final_train_loss": 980.7457598876953,
  "final_test_loss": 990.1863067626953,
  "total_epochs": 15,
  "best_train_loss": 980.7457598876953,
  "best_test_loss": 985.553515625
}