best val_rmse 0.0000

config.json

@@ -18,7 +18,7 @@
     "route": "oral",
     "time": 0.0
   },
-  "experiment_dir": "/work/ojedamarin/Projects/Pharma/Results/comet/node_pk_compartments/40a0407247cf4a4ba5af1dc855f2dd1e",
+  "experiment_dir": "/work/ojedamarin/Projects/Pharma/Results/comet/node_pk_compartments/23d894efece74fb091fcf88a59040fe0",
   "experiment_indentifier": null,
   "experiment_name": "node_pk_compartments",
   "hf_model_card_path": [
@@ -29,8 +29,8 @@
   "hugging_face_token": null,
   "meta_study": {
     "V_tmag_range": [
-      0.01,
-      0.1
+      0.001,
+      0.01
     ],
     "V_tscl_range": [
       1.0,
@@ -122,8 +122,8 @@
       3
     ],
     "rel_ruv_range": [
-      0.05,
-      0.3
+      0.02,
+      0.1
     ],
     "solver_method": "rk4",
     "time_num_steps": 100,
@@ -145,7 +145,7 @@
     ],
     "pretraining_epochs": 4800,
     "pretraining_protocol": "none",
-    "recreate_tempfile": false,
+    "recreate_tempfile": true,
     "return_split_versions": true,
     "split_seed": 42,
     "split_strategy": "study",
@@ -192,7 +192,7 @@
     "time_obs_encoder_hidden_dim": 256,
     "time_obs_encoder_output_dim": 256,
     "use_attention": true,
-    "use_kl_i": true,
+    "use_kl_i": false,
     "use_kl_init": true,
     "use_kl_s": true,
     "zi_latent_dim": 256