# configs/train_3photon.yaml
experiment:
  name: "3photon_12keV"

data:
  sample_folder: "/home/xie_x1/MLXID/DeepLearning/PileupSample"
  energy: 12 ### in keV

  batch_size_train: 4096
  batch_size_val: 8192
  batch_size_test: 8192
  num_workers: 16
  train_file_range: [0, 12]
  val_file_range: [13, 14]
  test_file_range: [15, 15]
  sample_ratio: 1.0
  n_size: 9 ### size of sub-images containing 3 photons

model:
  version: "260611" # 260529

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

loss:
  type: "three_point_set_loss_smooth_l1"
  smooth_l1_beta: 0.01