{
  "train_losses": [
    2100.769680175781,
    1210.195811767578,
    1176.4252709960938,
    1143.586368408203,
    1115.7994873046875,
    1091.0681024169921,
    1068.894476928711,
    1061.0595739746093,
    1048.5140423583985,
    1030.019833984375,
    1021.6400866699219,
    1014.914428100586,
    1012.0101983642578,
    1008.9616821289062,
    1007.0355267333985
  ],
  "test_losses": [
    1278.497021484375,
    1241.6509338378905,
    1210.9466491699218,
    1211.009942626953,
    1160.8113220214843,
    1111.9551544189453,
    1074.8253112792968,
    1102.0312622070312,
    1043.075909423828,
    1020.9971008300781,
    1005.1833709716797,
    1002.2283630371094,
    997.1159881591797,
    993.3500823974609,
    993.1695037841797
  ],
  "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": 10.0,
      "final_tc_beta": 10.0,
      "tc_beta_shape": "constant",
      "initial_dw_beta": 0.2,
      "final_dw_beta": 1.0,
      "dw_beta_shape": "custom",
      "warmup_epoch_ratio": 0.2,
      "free_bits": 0.15,
      "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": 10.0,
      "final_tc_beta": 10.0,
      "tc_beta_shape": "constant",
      "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.15,
      "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": 1007.0355267333985,
  "final_test_loss": 993.1695037841797,
  "total_epochs": 15,
  "best_train_loss": 1007.0355267333985,
  "best_test_loss": 993.1695037841797
}