File size: 2,303 Bytes
7509ffd 12413f5 7509ffd 12413f5 7509ffd 9fd4593 9fe6ce1 d119dd7 7509ffd d119dd7 9fe6ce1 49ce502 9fe6ce1 3416511 9fe6ce1 49ce502 9fe6ce1 7509ffd 49ce502 d119dd7 3416511 d119dd7 9fe6ce1 f19d2c2 49ce502 cb22ec8 80c3c17 9fe6ce1 d119dd7 bb958db d119dd7 bb958db d119dd7 7509ffd 12413f5 bb958db 12413f5 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": [
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
} |