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/d67e95712ee54b0899438bf7eb6311df",
+  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/ee2c92160537427cb3fc89f0056fd775",
   "experiment_indentifier": null,
   "experiment_name": "node_pk_compartments",
   "hf_model_card_path": [
@@ -189,6 +189,7 @@
     "node_step": true,
     "norm": "layer",
     "output_head_num_layers": 2,
+    "prediction_latent_deterministic": true,
     "prediction_only": true,
     "rnn_decoder_number_of_layers": 4,
     "rnn_individual_encoder_number_of_layers": 2,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:897d714b21ed6eb94f9d1f5a41ea077677a90e3d7856bc4ba8ec6a7c36e18113
+oid sha256:fff7a42d2f06d4d99e27bf442e394b55bcc0e566649b72376d113a66155d07ed
 size 6044367