From 3160d33c6f838ad81fd1a5b54af9833b4036cfa8 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Tue, 3 Sep 2024 22:34:17 +0100 Subject: [PATCH] Add debug lines --- src/training/histogram_dataset.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/src/training/histogram_dataset.py b/src/training/histogram_dataset.py index 9bf8f66..b9d526a 100644 --- a/src/training/histogram_dataset.py +++ b/src/training/histogram_dataset.py @@ -66,6 +66,7 @@ class HistogramDataset(Dataset): edited_histogram, original_histogram = self.read_2_histograms( cached_data_path, self._bin_count ) + logging.debug(f"Loaded {cached_data_path} from cache") except: logging.warning(f"Failed to load {cached_data_path}, regenerating...") else: @@ -80,11 +81,15 @@ class HistogramDataset(Dataset): ) if cached_data_path: - self.save_2_histograms( - edited_histogram, - original_histogram, - cached_data_path, - ) + try: + self.save_2_histograms( + edited_histogram, + original_histogram, + cached_data_path, + ) + logging.debug(f"Saved {cached_data_path} to cache") + except: + logging.warning(f"Failed to save {cached_data_path}") return ( torch.tensor(edited_histogram, dtype=torch.float).unsqueeze(0),