Add training data

training_data.json

@@ -1,9 +1,9 @@
 {
   "train_losses": [
-    3875.6037744140626
+    3751.7561962890627
   ],
   "test_losses": [
-    2741.24814453125
+    2578.7188354492187
   ],
   "config": {
     "hmm_only": false,
@@ -12,8 +12,8 @@
     "m_epochs_per_round": 1,
     "hmm_params": {
       "alpha": 5.0,
-      "kappa": 10.0,
-      "gamma": 10.0,
+      "kappa": 2.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": 3875.6037744140626,
-  "final_test_loss": 2741.24814453125,
+  "final_train_loss": 3751.7561962890627,
+  "final_test_loss": 2578.7188354492187,
   "total_epochs": 1,
-  "best_train_loss": 3875.6037744140626,
-  "best_test_loss": 2741.24814453125
+  "best_train_loss": 3751.7561962890627,
+  "best_test_loss": 2578.7188354492187
 }