# configs/train_1photon.yaml
experiment:
  name: "1photon_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
  num_aug_ops: 1
  normalize: false

  batch_size_train: 4096
  batch_size_val: 8192
  batch_size_test: 8192
  num_workers: 32
  train_file_range: [0, 12]
  val_file_range: [13, 14]
  test_file_range: [15, 15]

model:
  version: "260511" # 251022

training:
  epochs: 150
  learning_rate: 1.0e-3
  scheduler_factor: 0.7
  scheduler_patience: 3
  checkpoint_epochs: [10, 30, 50, 100, 150, 200, 300, 500, 1000]

loss:
  type: "huber_loss" 
  huber_beta: 0.01