CatkinChen
/

nethack-vae-hmm

Feature Extraction

MultiModalHackVAE_with_StickyHDPHMM

reinforcement-learning

variational-autoencoder

representation-learning

Model card Files Files and versions

CatkinChen commited on Sep 14

Commit

e7dd534

·

verified ·

1 Parent(s): c84fc7a

Add model config

Files changed (1) hide show

config.json +3 -3

config.json CHANGED Viewed

@@ -6,7 +6,7 @@
   "latent_dim": 96,
   "lowrank_dim": 0,
   "architecture": "Multi-modal Variational Autoencoder for NetHack game states",
-  "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": 3890.116960449219,
-  "final_test_loss": 2722.2674072265627,
   "m_step_epochs": 1
 }

   "latent_dim": 96,
   "lowrank_dim": 0,
   "architecture": "Multi-modal Variational Autoencoder for NetHack game states",
+  "round": 4,
   "total_rounds": 4,
   "hmm_parameters": {
     "alpha": 5.0,
   },
   "num_skills": 40,
   "base_vae_repo": "CatkinChen/nethack-vae",
+  "final_train_loss": 3825.648605957031,
+  "final_test_loss": 2584.886279296875,
   "m_step_epochs": 1
 }