Add training data

training_data.json

@@ -1,9 +1,9 @@
 {
   "train_losses": [
-    3865.235236816406
+    3861.2641748046876
   ],
   "test_losses": [
-    2759.7977783203123
+    2730.5551147460938
   ],
   "config": {
     "hmm_only": false,
@@ -24,12 +24,10 @@
       "nu0": 106
     },
     "hmm_paths": [
-      "checkpoints_hmm/hmm_round1.pt",
-      "checkpoints_hmm/hmm_round2.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_round1.pt"
     ],
     "hf_repos": {
       "hmm": "CatkinChen/nethack-hmm",
@@ -44,21 +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"
       }
     ]
   },
-  "final_train_loss": 3865.235236816406,
-  "final_test_loss": 2759.7977783203123,
+  "final_train_loss": 3861.2641748046876,
+  "final_test_loss": 2730.5551147460938,
   "total_epochs": 1,
-  "best_train_loss": 3865.235236816406,
-  "best_test_loss": 2759.7977783203123
+  "best_train_loss": 3861.2641748046876,
+  "best_test_loss": 2730.5551147460938
 }