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