From 786e256d90a4e9e4c3ff45f2c8fe4f974f3bfba9 Mon Sep 17 00:00:00 2001
From: rvandewater <r.p.vandewater@gmail.com>
Date: Tue, 15 Oct 2024 09:50:12 +0200
Subject: [PATCH] debugging

---
 icu_benchmarks/cross_validation.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/icu_benchmarks/cross_validation.py b/icu_benchmarks/cross_validation.py
index 3137821a..3a22f627 100644
--- a/icu_benchmarks/cross_validation.py
+++ b/icu_benchmarks/cross_validation.py
@@ -1,4 +1,5 @@
 import json
+import pickle
 from datetime import datetime
 import logging
 import gin
@@ -104,7 +105,10 @@ def execute_repeated_cv(
                 runmode=mode,
                 complete_train=complete_train
             )
-
+            # logging.debug(f"{data}")
+            # data_pickle_path = log_dir / "data.pkl"
+            # with open(data_pickle_path, "wb") as f:
+            #     pickle.dump(data, f, pickle.HIGHEST_PROTOCOL)
             preprocess_time = datetime.now() - start_time
             start_time = datetime.now()
             agg_loss += train_common(