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 7

Commit

ab8e49d

·

verified ·

1 Parent(s): cc23b65

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": 2,
   "total_rounds": 3,
   "hmm_parameters": {
     "alpha": 150.0,
@@ -23,7 +23,7 @@
   },
   "num_skills": 40,
   "base_vae_repo": "CatkinChen/nethack-vae",
-  "final_train_loss": 3865.235236816406,
-  "final_test_loss": 2759.7977783203123,
   "m_step_epochs": 1
 }

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