Update I/O and configs
This commit is contained in:
@@ -1,10 +1,10 @@
|
||||
# configs/train_1photon.yaml
|
||||
experiment:
|
||||
name: "1photon_12keV"
|
||||
name: "1photon_15keV"
|
||||
|
||||
data:
|
||||
sample_folder: "/mnt/sls_det_storage/moench_data/MLXID/Samples/Simulation/Moench040"
|
||||
energy: 12 ### in keV
|
||||
energy: 15 ### in keV
|
||||
noise_keV: 0.13
|
||||
noise_threshold: 0.0 ### set values below (noise * noise_threshold) to zero
|
||||
num_aug_ops: 1
|
||||
@@ -22,7 +22,7 @@ model:
|
||||
version: "260511" # 251022
|
||||
|
||||
training:
|
||||
epochs: 150
|
||||
epochs: 500
|
||||
learning_rate: 1.0e-3
|
||||
scheduler_factor: 0.7
|
||||
scheduler_patience: 3
|
||||
|
||||
@@ -1,10 +1,10 @@
|
||||
# configs/train_2photon.yaml
|
||||
experiment:
|
||||
name: "2photon_12keV"
|
||||
name: "2photon_15keV"
|
||||
|
||||
data:
|
||||
sample_folder: "/home/xie_x1/MLXID/DataProcess/Samples/SimuPileupSample/"
|
||||
energy: 12 ### in keV
|
||||
sample_folder: "/home/xie_x1/MLXID/DataProcess/Samples/15keV_SimuPileupSample/"
|
||||
energy: 15 ### in keV
|
||||
sample_ratio: 1.0
|
||||
noise_keV: 0.13
|
||||
|
||||
@@ -18,7 +18,7 @@ data:
|
||||
n_size: 6 ### size of sub-images containing 2 photons
|
||||
|
||||
model:
|
||||
version: "260608" ## 260608
|
||||
version: "260610" ### 260610 is the best one as of 260622
|
||||
|
||||
training:
|
||||
epochs: 1000
|
||||
|
||||
@@ -1,12 +1,12 @@
|
||||
# configs/train_3photon.yaml
|
||||
experiment:
|
||||
name: "3photon_12keV"
|
||||
name: "3photon_15keV"
|
||||
|
||||
data:
|
||||
sample_folder: "/home/xie_x1/MLXID/DeepLearning/PileupSample"
|
||||
energy: 12 ### in keV
|
||||
energy: 15 ### in keV
|
||||
|
||||
batch_size_train: 4096
|
||||
batch_size_train: 512
|
||||
batch_size_val: 8192
|
||||
batch_size_test: 8192
|
||||
num_workers: 16
|
||||
|
||||
Reference in New Issue
Block a user