best val_rmse 0.0000

config.json

@@ -18,7 +18,7 @@
     "route": "oral",
     "time": 0.0
   },
-  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/3783805d277346b4a3df0230611a6a22",
+  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/a339697c50314c81b3f21b24d326adad",
   "experiment_indentifier": null,
   "experiment_name": "node_pk_compartments",
   "hf_model_card_path": [
@@ -197,10 +197,10 @@
     "time_obs_encoder_output_dim": 128,
     "use_attention": true,
     "use_invariance_loss": true,
-    "use_kl_i": true,
-    "use_kl_i_np": true,
-    "use_kl_init": true,
-    "use_kl_s": true,
+    "use_kl_i": false,
+    "use_kl_i_np": false,
+    "use_kl_init": false,
+    "use_kl_s": false,
     "zi_latent_dim": 200
   },
   "run_index": 0,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2119e0f7063b4199a2d5f6440de1f8bcca42c1f81f6132016fef9d0b45905fed
+oid sha256:bf9f28181b16a651d082d47aa174d8d39ff583a80f38153ad6c098254c892ea4
 size 6044367