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,
|
|
@@ -28,12 +28,14 @@
|
|
| 28 |
"hmm_paths": [
|
| 29 |
"checkpoints_hmm/hmm_round1.pt",
|
| 30 |
"checkpoints_hmm/hmm_round2.pt",
|
| 31 |
-
"checkpoints_hmm/hmm_round3.pt"
|
|
|
|
| 32 |
],
|
| 33 |
"vae_hmm_paths": [
|
| 34 |
"checkpoints_hmm/vae_with_hmm_round1.pt",
|
| 35 |
"checkpoints_hmm/vae_with_hmm_round2.pt",
|
| 36 |
-
"checkpoints_hmm/vae_with_hmm_round3.pt"
|
|
|
|
| 37 |
],
|
| 38 |
"hf_repos": {
|
| 39 |
"hmm": "CatkinChen/nethack-hmm",
|
|
@@ -66,12 +68,21 @@
|
|
| 66 |
"skill_raster": "hmm_analysis/round_03/round03_skill_raster.png",
|
| 67 |
"dwell_pmfs": "hmm_analysis/round_03/round03_dwell_pmfs.png",
|
| 68 |
"diags_json": "hmm_analysis/round_03/round03_diags.json"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 69 |
}
|
| 70 |
]
|
| 71 |
},
|
| 72 |
-
"final_train_loss":
|
| 73 |
-
"final_test_loss":
|
| 74 |
"total_epochs": 1,
|
| 75 |
-
"best_train_loss":
|
| 76 |
-
"best_test_loss":
|
| 77 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"train_losses": [
|
| 3 |
+
3825.648605957031
|
| 4 |
],
|
| 5 |
"test_losses": [
|
| 6 |
+
2584.886279296875
|
| 7 |
],
|
| 8 |
"config": {
|
| 9 |
"hmm_only": false,
|
|
|
|
| 28 |
"hmm_paths": [
|
| 29 |
"checkpoints_hmm/hmm_round1.pt",
|
| 30 |
"checkpoints_hmm/hmm_round2.pt",
|
| 31 |
+
"checkpoints_hmm/hmm_round3.pt",
|
| 32 |
+
"checkpoints_hmm/hmm_round4.pt"
|
| 33 |
],
|
| 34 |
"vae_hmm_paths": [
|
| 35 |
"checkpoints_hmm/vae_with_hmm_round1.pt",
|
| 36 |
"checkpoints_hmm/vae_with_hmm_round2.pt",
|
| 37 |
+
"checkpoints_hmm/vae_with_hmm_round3.pt",
|
| 38 |
+
"checkpoints_hmm/vae_with_hmm_round4.pt"
|
| 39 |
],
|
| 40 |
"hf_repos": {
|
| 41 |
"hmm": "CatkinChen/nethack-hmm",
|
|
|
|
| 68 |
"skill_raster": "hmm_analysis/round_03/round03_skill_raster.png",
|
| 69 |
"dwell_pmfs": "hmm_analysis/round_03/round03_dwell_pmfs.png",
|
| 70 |
"diags_json": "hmm_analysis/round_03/round03_diags.json"
|
| 71 |
+
},
|
| 72 |
+
{
|
| 73 |
+
"dir": "hmm_analysis/round_04",
|
| 74 |
+
"pi_bar": "hmm_analysis/round_04/round04_pi_bar.png",
|
| 75 |
+
"A_heatmap": "hmm_analysis/round_04/round04_A_heatmap.png",
|
| 76 |
+
"mu_pca": "hmm_analysis/round_04/round04_mu_t-sne.png",
|
| 77 |
+
"skill_raster": "hmm_analysis/round_04/round04_skill_raster.png",
|
| 78 |
+
"dwell_pmfs": "hmm_analysis/round_04/round04_dwell_pmfs.png",
|
| 79 |
+
"diags_json": "hmm_analysis/round_04/round04_diags.json"
|
| 80 |
}
|
| 81 |
]
|
| 82 |
},
|
| 83 |
+
"final_train_loss": 3825.648605957031,
|
| 84 |
+
"final_test_loss": 2584.886279296875,
|
| 85 |
"total_epochs": 1,
|
| 86 |
+
"best_train_loss": 3825.648605957031,
|
| 87 |
+
"best_test_loss": 2584.886279296875
|
| 88 |
}
|