Add model config

config.json

@@ -3,7 +3,7 @@
   "framework": "PyTorch",
   "task": "representation-learning",
   "dataset": "NetHack Learning Dataset",
-  "latent_dim": 128,
+  "latent_dim": 96,
   "lowrank_dim": 0,
   "bInclude_glyph_bag": true,
   "bInclude_hero": true,
@@ -58,9 +58,9 @@
       "best_epoch": null
     }
   },
-  "final_train_loss": 344.93716651916503,
-  "final_test_loss": 1596.8008026123048,
-  "best_train_loss": 344.93716651916503,
-  "best_test_loss": 384.25682220458987,
+  "final_train_loss": 347.81424446105956,
+  "final_test_loss": 1307.968310546875,
+  "best_train_loss": 347.81424446105956,
+  "best_test_loss": 371.5667358398438,
   "total_epochs": 15
 }