best val_rmse 0.0434

config.json

@@ -1,5 +1,5 @@
 {
-  "best_val_loss": 0.16986101865768433,
+  "best_val_loss": 0.043431881815195084,
   "comet_ai_key": null,
   "context_observations": {
     "add_rem": true,
@@ -33,7 +33,7 @@
     "same_route": true,
     "time": 0.0
   },
-  "experiment_dir": "/home/cesarali/Pharma/sim_priors_pk/results/comet/context-vae-pk/3ec2b7b8ff714f488fbf0a12a977b2b2",
+  "experiment_dir": "/home/cesarali/Pharma/sim_priors_pk/results/comet/context-vae-pk/ace5227fd3f4472f934a2d5e6b4bb13b",
   "experiment_indentifier": null,
   "experiment_name": "context-vae-pk",
   "hf_model_card_path": [
@@ -44,12 +44,16 @@
   "hugging_face_token": null,
   "meta_study": {
     "band_scale_range": [
-      0.1,
-      0.3
+      0.05,
+      1.0
     ],
     "baseline_range": [
-      0.0,
-      0.1
+      0.6,
+      1.4
+    ],
+    "decay_rate_range": [
+      0.001,
+      0.2
     ],
     "drug_id_options": [
       "DummyDrug"
@@ -59,7 +63,11 @@
       16,
       16
     ],
-    "seed": 123,
+    "num_peripherals_range": [
+      1,
+      3
+    ],
+    "p1": 0.5,
     "simple_mode": true,
     "solver_method": "dummy",
     "time_num_steps": 40,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:160a4eb2cc140d831c090b4e29370cea50be7a7c547699498a1750fc4436cbf6
+oid sha256:2c8b2a4d5691db9e97eba70e10fc53ba26f7971ec391fd29d86cf45384604ff2
 size 39801963