best val_rmse 0.0712

config.json

@@ -1,5 +1,5 @@
 {
-  "best_val_loss": 0.06213032454252243,
+  "best_val_loss": 0.07117409259080887,
   "comet_ai_key": null,
   "context_observations": {
     "add_rem": true,
@@ -17,7 +17,7 @@
     "route": "oral",
     "time": 0.0
   },
-  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/b69aa8087a7148b1a65a387730b35161",
+  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/faf07144e8404dd0948c02a400b7fdc5",
   "experiment_indentifier": null,
   "experiment_name": "node_pk_compartments",
   "hf_model_card_path": [
@@ -188,7 +188,7 @@
       0.9,
       0.999
     ],
-    "epochs": 5,
+    "epochs": 3,
     "eps": 1e-08,
     "gradient_clip_val": 1.0,
     "learning_rate": 0.0001,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:881ced016b6eb928b492220ab32ad2a18936884050dd80e93ba3fa76f9091f86
+oid sha256:b39f6b7dca9c5c341b94ba380aab0b6a0ff48638282e981da692b42a2c19ccd6
 size 4910617