36 lines
744 B
YAML
36 lines
744 B
YAML
# 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: 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]
|
|
|
|
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 |