File size: 2,532 Bytes
3481865 8533274 3481865 8533274 3481865 82a9d0a 52156bb c6fa805 3481865 4f6d369 49cd5e2 8d549a7 3481865 a132d44 661e36f a132d44 3481865 82a9d0a 51710a2 1a9bb2d 8533274 82a9d0a 3481865 51710a2 1a9bb2d 8533274 3481865 82a9d0a 51710a2 1a9bb2d 82a9d0a 3481865 8533274 c6fa805 8533274 3481865 |
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 |
{
"train_losses": [
3835.708234863281
],
"test_losses": [
2622.4922485351562
],
"config": {
"hmm_only": false,
"vae_only_with_hmm": false,
"em_rounds": 4,
"m_epochs_per_round": 1,
"hmm_params": {
"alpha": 5.0,
"kappa": 1.0,
"gamma": 5.0,
"K": 40,
"D": 96
},
"niw_params": {
"mu0": 0.0,
"kappa0": 1.0,
"Psi0": 30.0,
"nu0": 106,
"set_mu0_with_global_mean": false,
"set_Psi0_with_global_cov": false
},
"hmm_paths": [
"checkpoints_hmm/hmm_round1.pt",
"checkpoints_hmm/hmm_round2.pt",
"checkpoints_hmm/hmm_round3.pt"
],
"vae_hmm_paths": [
"checkpoints_hmm/vae_with_hmm_round1.pt",
"checkpoints_hmm/vae_with_hmm_round2.pt",
"checkpoints_hmm/vae_with_hmm_round3.pt"
],
"hf_repos": {
"hmm": "CatkinChen/nethack-hmm",
"vae_hmm": "CatkinChen/nethack-vae-hmm"
},
"viz_paths": [
{
"dir": "hmm_analysis/round_01",
"pi_bar": "hmm_analysis/round_01/round01_pi_bar.png",
"A_heatmap": "hmm_analysis/round_01/round01_A_heatmap.png",
"mu_pca": "hmm_analysis/round_01/round01_mu_t-sne.png",
"skill_raster": "hmm_analysis/round_01/round01_skill_raster.png",
"dwell_pmfs": "hmm_analysis/round_01/round01_dwell_pmfs.png",
"diags_json": "hmm_analysis/round_01/round01_diags.json"
},
{
"dir": "hmm_analysis/round_02",
"pi_bar": "hmm_analysis/round_02/round02_pi_bar.png",
"A_heatmap": "hmm_analysis/round_02/round02_A_heatmap.png",
"mu_pca": "hmm_analysis/round_02/round02_mu_t-sne.png",
"skill_raster": "hmm_analysis/round_02/round02_skill_raster.png",
"dwell_pmfs": "hmm_analysis/round_02/round02_dwell_pmfs.png",
"diags_json": "hmm_analysis/round_02/round02_diags.json"
},
{
"dir": "hmm_analysis/round_03",
"pi_bar": "hmm_analysis/round_03/round03_pi_bar.png",
"A_heatmap": "hmm_analysis/round_03/round03_A_heatmap.png",
"mu_pca": "hmm_analysis/round_03/round03_mu_t-sne.png",
"skill_raster": "hmm_analysis/round_03/round03_skill_raster.png",
"dwell_pmfs": "hmm_analysis/round_03/round03_dwell_pmfs.png",
"diags_json": "hmm_analysis/round_03/round03_diags.json"
}
]
},
"final_train_loss": 3835.708234863281,
"final_test_loss": 2622.4922485351562,
"total_epochs": 1,
"best_train_loss": 3835.708234863281,
"best_test_loss": 2622.4922485351562
} |