best val_rmse 0.0002

config.json

@@ -1,5 +1,5 @@
 {
-  "best_val_loss": 0.0001485223474446684,
+  "best_val_loss": 0.00022563124366570264,
   "comet_ai_key": null,
   "context_observations": {
     "add_rem": true,
@@ -17,7 +17,7 @@
     "route": "oral",
     "time": 0.0
   },
-  "experiment_dir": null,
+  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/457ede8087e74c5bbc20a2a6fa3645aa",
   "experiment_indentifier": null,
   "experiment_name": "node_pk_compartments",
   "hf_model_card_path": [
@@ -119,6 +119,7 @@
     ],
     "pretraining_epochs": 90,
     "pretraining_protocol": "none",
+    "return_split_versions": false,
     "split_seed": 42,
     "split_strategy": "subject",
     "test_protocol": "empirical",
@@ -136,6 +137,7 @@
     "aggregator_num_heads": 8,
     "aggregator_type": "attention",
     "cov_proj_dim": 16,
+    "decoder_attention_layers": 2,
     "decoder_hidden_dim": 32,
     "decoder_name": "RNNDecoder",
     "decoder_num_layers": 2,
@@ -158,6 +160,9 @@
     "time_obs_encoder_hidden_dim": 32,
     "time_obs_encoder_output_dim": 32,
     "use_attention": true,
+    "use_kl_i": true,
+    "use_kl_init": true,
+    "use_kl_s": true,
     "zi_latent_dim": 20
   },
   "run_index": 0,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:933b8e140ae5023540a8d5cb7821b25f22e828e34cdde4b937cb7466f2ae9ab2
+oid sha256:7213605d3be830cd8d0be3151914f98affb0ca21e2ea634c3994533efcdb1bec
 size 116067