Update 2-ph training to a modern style

Co-authored-by: Copilot <copilot@github.com>
2026-05-06 16:33:01 +02:00
parent d9917d7996
commit 66a0fbffcc
2 changed files with 201 additions and 0 deletions
@@ -0,0 +1,33 @@
+# configs/train_2photon.yaml
+experiment:
+  name: "2photon_12keV"
+
+data:
+  sample_folder: "/mnt/sls_det_storage/moench_data/MLXID/Samples/Simulation/Moench040"
+  energy: 12 ### in keV
+  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
+  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
+
+model:
+  version: "251124"
+
+training:
+  epochs: 150
+  learning_rate: 1.0e-3
+  weight_decay: 1.0e-4
+  scheduler_factor: 0.7
+  scheduler_patience: 5
+  checkpoint_epochs: [10, 30, 50, 100, 150]
+
+loss:
+  type: "two_point_set_loss_l2"