best val_rmse 0.0000

config.json

@@ -12,13 +12,13 @@
     "obs_type": "observations_pk_peak_halflife",
     "past_time_ratio": 0.1
   },
-  "debug_test": false,
+  "debug_test": true,
   "dosing": {
     "dose": 1.0,
     "route": "oral",
     "time": 0.0
   },
-  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/0a6d86a683b44ebaa1c3f31353476d4b",
+  "experiment_dir": "/home/cesarali/Pharma/generative_pk/results/comet/node_pk_compartments/8a3c590fc93645fc80a04e29e8e91916",
   "experiment_indentifier": null,
   "experiment_name": "node_pk_compartments",
   "hf_model_card_path": [
@@ -149,7 +149,7 @@
     "return_split_versions": true,
     "split_seed": 42,
     "split_strategy": "study",
-    "store_in_tempfile": true,
+    "store_in_tempfile": false,
     "tempfile_path": [
       "preprocessed",
       "simulated_ou_as_rates"
@@ -185,7 +185,7 @@
     "individual_encoder_number_of_heads": 4,
     "init_hidden_num_layers": 2,
     "input_encoding_hidden_dim": 128,
-    "loss_name": "nll",
+    "loss_name": "log_nll",
     "node_step": true,
     "norm": "layer",
     "output_head_num_layers": 2,
@@ -230,7 +230,7 @@
     ],
     "epochs": 100,
     "eps": 1e-08,
-    "eval_empirical_every_pct": 0.25,
+    "eval_empirical_every_pct": 0.5,
     "gradient_clip_val": 1.0,
     "learning_rate": 0.0001,
     "log_image_every_epoch": 2,

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7833e1a200e3070d803d3901dff9a9a78092ced8de904698c29847342e679208
+oid sha256:5834d50496a7ffd314012ff0dfa0648fa3a89b46711280876e432736bf24c77a
 size 6044367