Add training data

training_data.json

@@ -1,9 +1,9 @@
 {
   "train_losses": [
-    3844.8290478515623
+    3863.50556640625
   ],
   "test_losses": [
-    2720.714489746094
+    2759.2317626953127
   ],
   "config": {
     "hmm_only": false,
@@ -12,7 +12,7 @@
     "m_epochs_per_round": 1,
     "hmm_params": {
       "alpha": 5.0,
-      "kappa": 2.0,
+      "kappa": 1.0,
       "gamma": 5.0,
       "K": 40,
       "D": 96
@@ -24,14 +24,10 @@
       "nu0": 106
     },
     "hmm_paths": [
-      "checkpoints_hmm/hmm_round1.pt",
-      "checkpoints_hmm/hmm_round2.pt",
-      "checkpoints_hmm/hmm_round3.pt"
+      "checkpoints_hmm/hmm_round1.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"
+      "checkpoints_hmm/vae_with_hmm_round1.pt"
     ],
     "hf_repos": {
       "hmm": "CatkinChen/nethack-hmm",
@@ -46,30 +42,12 @@
         "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": 3844.8290478515623,
-  "final_test_loss": 2720.714489746094,
+  "final_train_loss": 3863.50556640625,
+  "final_test_loss": 2759.2317626953127,
   "total_epochs": 1,
-  "best_train_loss": 3844.8290478515623,
-  "best_test_loss": 2720.714489746094
+  "best_train_loss": 3863.50556640625,
+  "best_test_loss": 2759.2317626953127
 }