# 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: "260507" ## 251124 training: epochs: 1000 learning_rate: 1.0e-3 weight_decay: 1.0e-4 scheduler_factor: 0.7 scheduler_patience: 5 checkpoint_epochs: [10, 30, 50, 100, 150, 300, 500, 1000] loss: type: "two_point_set_loss_smooth_l1" #"two_point_set_loss_l2"