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 15

Commit

a2bfb2f

·

verified ·

1 Parent(s): 1d79e13

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": 1,
   "total_rounds": 4,
   "hmm_parameters": {
     "alpha": 5.0,
@@ -25,7 +25,7 @@
   },
   "num_skills": 40,
   "base_vae_repo": "CatkinChen/nethack-vae",
-  "final_train_loss": 3815.7067602539064,
-  "final_test_loss": 2713.068896484375,
   "m_step_epochs": 1
 }

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