Reframe the training codes, add yaml config

2026-03-20 11:06:53 +01:00
parent 96e78f8ab5
commit 90cb2f5a47
2 changed files with 165 additions and 119 deletions
@@ -0,0 +1,40 @@
+# configs/train_1photon.yaml
+experiment:
+  name: "1photon_15keV"
+
+data:
+  sample_folder: "/mnt/sls_det_storage/moench_data/MLXID/Samples/Simulation/Moench040"
+  energy: 15 ### in keV
+  noise_keV: 0.13
+  noise_threshold: 0.0 ### set values below (noise * noise_threshold) to zero
+  num_aug_ops: 1
+  normalize: true
+
+  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]
+
+model:
+  version: "251022"
+
+training:
+  epochs: 15
+  learning_rate: 1.0e-3
+  loss_alpha: 7.0
+  loss_beta: 3.0
+  scheduler_factor: 0.7
+  scheduler_patience: 3
+  checkpoint_epochs: [10, 30, 50, 100, 150]
+
+loss:
+  type: "weightedL1"
+  alpha: 7.0
+  beta: 3.0
+
+logging:
+  save_dir: "Results"
+  model_dir: "Models"