Add training data
Browse files- training_data.json +19 -8
training_data.json
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
{
|
| 2 |
"train_losses": [
|
| 3 |
-
|
| 4 |
],
|
| 5 |
"test_losses": [
|
| 6 |
-
|
| 7 |
],
|
| 8 |
"config": {
|
| 9 |
"hmm_only": false,
|
|
@@ -26,10 +26,12 @@
|
|
| 26 |
"set_Psi0_with_global_cov": false
|
| 27 |
},
|
| 28 |
"hmm_paths": [
|
| 29 |
-
"checkpoints_hmm/hmm_round1.pt"
|
|
|
|
| 30 |
],
|
| 31 |
"vae_hmm_paths": [
|
| 32 |
-
"checkpoints_hmm/vae_with_hmm_round1.pt"
|
|
|
|
| 33 |
],
|
| 34 |
"hf_repos": {
|
| 35 |
"hmm": "CatkinChen/nethack-hmm",
|
|
@@ -44,12 +46,21 @@
|
|
| 44 |
"skill_raster": "hmm_analysis/round_01/round01_skill_raster.png",
|
| 45 |
"dwell_pmfs": "hmm_analysis/round_01/round01_dwell_pmfs.png",
|
| 46 |
"diags_json": "hmm_analysis/round_01/round01_diags.json"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 47 |
}
|
| 48 |
]
|
| 49 |
},
|
| 50 |
-
"final_train_loss":
|
| 51 |
-
"final_test_loss":
|
| 52 |
"total_epochs": 1,
|
| 53 |
-
"best_train_loss":
|
| 54 |
-
"best_test_loss":
|
| 55 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"train_losses": [
|
| 3 |
+
3869.1480883789063
|
| 4 |
],
|
| 5 |
"test_losses": [
|
| 6 |
+
2721.682666015625
|
| 7 |
],
|
| 8 |
"config": {
|
| 9 |
"hmm_only": false,
|
|
|
|
| 26 |
"set_Psi0_with_global_cov": false
|
| 27 |
},
|
| 28 |
"hmm_paths": [
|
| 29 |
+
"checkpoints_hmm/hmm_round1.pt",
|
| 30 |
+
"checkpoints_hmm/hmm_round2.pt"
|
| 31 |
],
|
| 32 |
"vae_hmm_paths": [
|
| 33 |
+
"checkpoints_hmm/vae_with_hmm_round1.pt",
|
| 34 |
+
"checkpoints_hmm/vae_with_hmm_round2.pt"
|
| 35 |
],
|
| 36 |
"hf_repos": {
|
| 37 |
"hmm": "CatkinChen/nethack-hmm",
|
|
|
|
| 46 |
"skill_raster": "hmm_analysis/round_01/round01_skill_raster.png",
|
| 47 |
"dwell_pmfs": "hmm_analysis/round_01/round01_dwell_pmfs.png",
|
| 48 |
"diags_json": "hmm_analysis/round_01/round01_diags.json"
|
| 49 |
+
},
|
| 50 |
+
{
|
| 51 |
+
"dir": "hmm_analysis/round_02",
|
| 52 |
+
"pi_bar": "hmm_analysis/round_02/round02_pi_bar.png",
|
| 53 |
+
"A_heatmap": "hmm_analysis/round_02/round02_A_heatmap.png",
|
| 54 |
+
"mu_pca": "hmm_analysis/round_02/round02_mu_t-sne.png",
|
| 55 |
+
"skill_raster": "hmm_analysis/round_02/round02_skill_raster.png",
|
| 56 |
+
"dwell_pmfs": "hmm_analysis/round_02/round02_dwell_pmfs.png",
|
| 57 |
+
"diags_json": "hmm_analysis/round_02/round02_diags.json"
|
| 58 |
}
|
| 59 |
]
|
| 60 |
},
|
| 61 |
+
"final_train_loss": 3869.1480883789063,
|
| 62 |
+
"final_test_loss": 2721.682666015625,
|
| 63 |
"total_epochs": 1,
|
| 64 |
+
"best_train_loss": 3869.1480883789063,
|
| 65 |
+
"best_test_loss": 2721.682666015625
|
| 66 |
}
|