revert last redundant vae hmm training

config.json

@@ -6,7 +6,7 @@
   "latent_dim": 96,
   "lowrank_dim": 0,
   "architecture": "Multi-modal Variational Autoencoder for NetHack game states",
-  "round": 4,
+  "round": 3,
   "total_rounds": 4,
   "hmm_parameters": {
     "alpha": 5.0,
@@ -25,7 +25,7 @@
   },
   "num_skills": 40,
   "base_vae_repo": "CatkinChen/nethack-vae",
-  "final_train_loss": 3909.512939453125,
-  "final_test_loss": 2831.5083740234377,
+  "final_train_loss": 3835.708234863281,
+  "final_test_loss": 2622.4922485351562,
   "m_step_epochs": 1
 }

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b09fc2dfdca36e43fdc767a55497509a3b8944297f3ea5d7333bd1a3974329f9
+oid sha256:3ccaeb422e83eda57fe8dd855ab068bd11fd39b9795bc6a0d58bca8e55d8f5a9
 size 9364123

training_data.json

@@ -1,9 +1,9 @@
 {
   "train_losses": [
-    3909.512939453125
+    3835.708234863281
   ],
   "test_losses": [
-    2831.5083740234377
+    2622.4922485351562
   ],
   "config": {
     "hmm_only": false,
@@ -28,14 +28,12 @@
     "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_round3.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_round3.pt"
     ],
     "hf_repos": {
       "hmm": "CatkinChen/nethack-hmm",
@@ -68,21 +66,12 @@
         "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": 3909.512939453125,
-  "final_test_loss": 2831.5083740234377,
+  "final_train_loss": 3835.708234863281,
+  "final_test_loss": 2622.4922485351562,
   "total_epochs": 1,
-  "best_train_loss": 3909.512939453125,
-  "best_test_loss": 2831.5083740234377
+  "best_train_loss": 3835.708234863281,
+  "best_test_loss": 2622.4922485351562
 }

vae_with_hmm_round4.pt

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:3cf3ac4dcdcaf14ac7669402670c64ec2af7d3fb91e762d24d4423ff6d3bc4da
-size 10955067
+oid sha256:557ceb7e11b49124cd468ac2af8f8d54ee26d73cbe24946cfc4acb8f2d21b497
+size 10951931