MLXID/DeepLearning
0
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update I/O and configs

Browse Source
This commit is contained in:
xie_x1 committed 2026-08-21 09:11:38 +02:00
1 parent 7c7dc28ce0
commit 3686b3e8c4
5 files changed
+16 -16
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

Train_2Photon.py
+1 -1
View File
@@ -181,7 +181,7 @@ def get_dataloaders(conf):
file_range_keys = ['train_file_range', 'val_file_range', 'test_file_range']
for split, key, batch_key, file_range_key in zip(splits, keys, batch_keys, file_range_keys):
files = [f"{conf.data.sample_folder}/pileupOf2phs_sample_{i}.npz" for i in range(conf.data[file_range_key][0], conf.data[file_range_key][1] + 1)]
files = [f"{conf.data.sample_folder}/{conf.data.energy}keV_Noise{conf.data.noise_keV}keV_pileupOf2phs_sample_{i}.npz" for i in range(conf.data[file_range_key][0], conf.data[file_range_key][1] + 1)]
datasets[split] = doublePhotonDataset(
files,