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,
|
|
@@ -27,11 +27,13 @@
|
|
| 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",
|
|
@@ -55,12 +57,21 @@
|
|
| 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":
|
| 62 |
-
"final_test_loss":
|
| 63 |
"total_epochs": 1,
|
| 64 |
-
"best_train_loss":
|
| 65 |
-
"best_test_loss":
|
| 66 |
}
|
|
|
|
| 1 |
{
|
| 2 |
"train_losses": [
|
| 3 |
+
3890.116960449219
|
| 4 |
],
|
| 5 |
"test_losses": [
|
| 6 |
+
2722.2674072265627
|
| 7 |
],
|
| 8 |
"config": {
|
| 9 |
"hmm_only": false,
|
|
|
|
| 27 |
},
|
| 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",
|
|
|
|
| 57 |
"skill_raster": "hmm_analysis/round_02/round02_skill_raster.png",
|
| 58 |
"dwell_pmfs": "hmm_analysis/round_02/round02_dwell_pmfs.png",
|
| 59 |
"diags_json": "hmm_analysis/round_02/round02_diags.json"
|
| 60 |
+
},
|
| 61 |
+
{
|
| 62 |
+
"dir": "hmm_analysis/round_03",
|
| 63 |
+
"pi_bar": "hmm_analysis/round_03/round03_pi_bar.png",
|
| 64 |
+
"A_heatmap": "hmm_analysis/round_03/round03_A_heatmap.png",
|
| 65 |
+
"mu_pca": "hmm_analysis/round_03/round03_mu_t-sne.png",
|
| 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": 3890.116960449219,
|
| 73 |
+
"final_test_loss": 2722.2674072265627,
|
| 74 |
"total_epochs": 1,
|
| 75 |
+
"best_train_loss": 3890.116960449219,
|
| 76 |
+
"best_test_loss": 2722.2674072265627
|
| 77 |
}
|