# configs/train_2photon.yaml
experiment:
  name: "2photon_12keV"

data:
  sample_folder: "/home/xie_x1/MLXID/DataProcess/Samples/SimuPileupSample/"
  energy: 12 ### in keV
  sample_ratio: 1.0
  noise_keV: 0.13

  batch_size_train: 4096
  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: 6 ### size of sub-images containing 2 photons

model:
  version: "260608" ## 260608

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

loss:
  type: "two_point_set_loss_smooth_l1" #"two_point_set_loss_l2", "two_point_set_loss_smooth_l1"
  huber_beta: 0.01