Update 2ph training config
This commit is contained in:
+11
-11
@@ -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
|
||||
Reference in New Issue
Block a user