best val_rmse 0.1920

config.json

@@ -1,5 +1,5 @@
 {
-  "best_val_loss": 0.18817822635173798,
+  "best_val_loss": 0.1920255869626999,
   "comet_ai_key": null,
   "context_observations": {
     "add_rem": true,
@@ -18,7 +18,7 @@
     "route": "oral",
     "time": 0.0
   },
-  "experiment_dir": "/work/ojedamarin/Projects/Pharma/Results/comet/node_pk_compartments/e2d5859dc0cf4ffc929c01783e900495",
+  "experiment_dir": "/work/ojedamarin/Projects/Pharma/Results/comet/node_pk_compartments/11ea6ff2323e4cf7b9583a578457dfcb",
   "experiment_indentifier": null,
   "experiment_name": "node_pk_compartments",
   "hf_model_card_path": [
@@ -195,9 +195,9 @@
     "use_kl_i": true,
     "use_kl_init": true,
     "use_kl_s": true,
-    "zi_latent_dim": 25
+    "zi_latent_dim": 100
   },
-  "run_index": 0,
+  "run_index": 1,
   "tags": [
     "NP",
     "OU",

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d880b6355b7cb46f124ad047f1db6f10c1b07a211c1dd9f0084ca974e55f24b8
-size 35154118
+oid sha256:c6b9f2f57378f3ffb236ef4ee5fd0cbaff7c5dfbcb9ceded5f69fdb0412026ec
+size 35650566