File size: 2,286 Bytes
7509ffd 3bdd0a9 7509ffd 3bdd0a9 7509ffd 9fd4593 9fe6ce1 d119dd7 7509ffd d119dd7 9fe6ce1 49ce502 9fe6ce1 3bdd0a9 3416511 3bdd0a9 9fe6ce1 49ce502 9fe6ce1 3bdd0a9 9fe6ce1 7509ffd 49ce502 d119dd7 3bdd0a9 3416511 3bdd0a9 9fe6ce1 f19d2c2 49ce502 cb22ec8 80c3c17 9fe6ce1 3bdd0a9 9fe6ce1 d119dd7 bb958db d119dd7 bb958db d119dd7 7509ffd 3bdd0a9 bb958db 3bdd0a9 7509ffd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
{
"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
} |