best val_rmse 0.4955

config.json

@@ -1,5 +1,5 @@
 {
-  "best_val_loss": 0.2617513835430145,
+  "best_val_loss": 0.4954600930213928,
   "comet_ai_key": null,
   "context_observations": {
     "add_rem": true,
@@ -18,7 +18,7 @@
     "route": "oral",
     "time": 0.0
   },
-  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/0f6071c9a85d4827abcd8ef946a7428a",
+  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/778551c8960c4b158a408845370c8661",
   "experiment_indentifier": null,
   "experiment_name": "node_pk_compartments",
   "hf_model_card_path": [
@@ -220,7 +220,7 @@
       0.9,
       0.999
     ],
-    "epochs": 3,
+    "epochs": 30,
     "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:07a25afe0f0e94cbf32f3f70f4297bbf1ac46501b1ed3f47379fa65b6ec0c0ea
+oid sha256:af1d074a5e29a6239d98a876b491cc914cb4a6101ee6d82d53a3131aa830b74d
 size 168701