andras/bipolaroidbipolaroid
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add debug lines

Browse source
This commit is contained in:
Andras Schmelczer 2024-09-03 22:34:17 +01:00
parent edd92593b7
commit 3160d33c6f
No known key found for this signature in database
GPG key ID: FC8F2C3D3D1A718C
1 changed files with 10 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/training/histogram_dataset.py
View file

@ -66,6 +66,7 @@ class HistogramDataset(Dataset):
edited_histogram, original_histogram = self.read_2_histograms( edited_histogram, original_histogram = self.read_2_histograms(
cached_data_path, self._bin_count cached_data_path, self._bin_count
) )
logging.debug(f"Loaded {cached_data_path} from cache")
except: except:
logging.warning(f"Failed to load {cached_data_path}, regenerating...") logging.warning(f"Failed to load {cached_data_path}, regenerating...")
else: else:
@ -80,11 +81,15 @@ class HistogramDataset(Dataset):
) )
if cached_data_path: if cached_data_path:
try:
self.save_2_histograms( self.save_2_histograms(
edited_histogram, edited_histogram,
original_histogram, original_histogram,
cached_data_path, cached_data_path,
) )
logging.debug(f"Saved {cached_data_path} to cache")
except:
logging.warning(f"Failed to save {cached_data_path}")
return ( return (
torch.tensor(edited_histogram, dtype=torch.float).unsqueeze(0), torch.tensor(edited_histogram, dtype=torch.float).unsqueeze(0),