diff --git a/Configs/train_2photon.yaml b/Configs/train_2photon.yaml
index 9bc79e9..f6878ab 100644
--- a/Configs/train_2photon.yaml
+++ b/Configs/train_2photon.yaml
@@ -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"
\ No newline at end of file
+  type: "two_point_set_loss_smooth_l1" #"two_point_set_loss_l2", "two_point_set_loss_smooth_l1"
+  huber_beta: 0.01
\ No newline at end of file