Update 2ph training config

This commit is contained in:
2026-06-10 11:26:48 +02:00
parent 9a38cdbe8e
commit 2ef6f79480
+11 -11
View File
@@ -3,31 +3,31 @@ experiment:
name: "2photon_12keV"
data:
sample_folder: "/mnt/sls_det_storage/moench_data/MLXID/Samples/Simulation/Moench040"
sample_folder: "/home/xie_x1/MLXID/DataProcess/Samples/SimuPileupSample/"
energy: 12 ### in keV
sample_ratio: 1.0
noise_keV: 0.13
noise_threshold: 0.0 ### set values below (noise * noise_threshold) to zero
normalize: false
batch_size_train: 4096
batch_size_val: 1024
batch_size_test: 1024
num_workers: 32
batch_size_val: 8196
batch_size_test: 8196
num_workers: 16
train_file_range: [0, 12]
val_file_range: [13, 14]
test_file_range: [15, 15]
n_size: 7 ### size of sub-images containing 2 photons
n_size: 6 ### size of sub-images containing 2 photons
model:
version: "260507" ## 251124
version: "260608" ## 260608
training:
epochs: 1000
learning_rate: 1.0e-3
weight_decay: 1.0e-4
weight_decay: 0
scheduler_factor: 0.7
scheduler_patience: 5
checkpoint_epochs: [10, 30, 50, 100, 150, 300, 500, 1000]
checkpoint_epochs: [10, 30, 50, 100, 150, 300, 500]
loss:
type: "two_point_set_loss_smooth_l1" #"two_point_set_loss_l2"
type: "two_point_set_loss_smooth_l1" #"two_point_set_loss_l2", "two_point_set_loss_smooth_l1"
huber_beta: 0.01