Add training data

training_data.json

@@ -1,9 +1,9 @@
 {
   "train_losses": [
-    3825.648605957031
+    3815.7067602539064
   ],
   "test_losses": [
-    2584.886279296875
+    2713.068896484375
   ],
   "config": {
     "hmm_only": false,
@@ -26,16 +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_round4.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_round4.pt"
+      "checkpoints_hmm/vae_with_hmm_round1.pt"
     ],
     "hf_repos": {
       "hmm": "CatkinChen/nethack-hmm",
@@ -50,39 +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"
-      },
-      {
-        "dir": "hmm_analysis/round_04",
-        "pi_bar": "hmm_analysis/round_04/round04_pi_bar.png",
-        "A_heatmap": "hmm_analysis/round_04/round04_A_heatmap.png",
-        "mu_pca": "hmm_analysis/round_04/round04_mu_t-sne.png",
-        "skill_raster": "hmm_analysis/round_04/round04_skill_raster.png",
-        "dwell_pmfs": "hmm_analysis/round_04/round04_dwell_pmfs.png",
-        "diags_json": "hmm_analysis/round_04/round04_diags.json"
       }
     ]
   },
-  "final_train_loss": 3825.648605957031,
-  "final_test_loss": 2584.886279296875,
+  "final_train_loss": 3815.7067602539064,
+  "final_test_loss": 2713.068896484375,
   "total_epochs": 1,
-  "best_train_loss": 3825.648605957031,
-  "best_test_loss": 2584.886279296875
+  "best_train_loss": 3815.7067602539064,
+  "best_test_loss": 2713.068896484375
 }