Add training data

training_data.json

@@ -1,14 +1,14 @@
 {
   "train_losses": [
-    3866.4689331054688
+    3848.3857275390624
   ],
   "test_losses": [
-    2702.8141845703126
+    2673.374951171875
   ],
   "config": {
     "hmm_only": false,
     "vae_only_with_hmm": false,
-    "em_rounds": 3,
+    "em_rounds": 4,
     "m_epochs_per_round": 1,
     "hmm_params": {
       "alpha": 5.0,
@@ -26,14 +26,10 @@
       "set_Psi0_with_global_cov": false
     },
     "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",
@@ -48,30 +44,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": 3866.4689331054688,
-  "final_test_loss": 2702.8141845703126,
+  "final_train_loss": 3848.3857275390624,
+  "final_test_loss": 2673.374951171875,
   "total_epochs": 1,
-  "best_train_loss": 3866.4689331054688,
-  "best_test_loss": 2702.8141845703126
+  "best_train_loss": 3848.3857275390624,
+  "best_test_loss": 2673.374951171875
 }