Update train yaml

This commit is contained in:
2026-03-20 16:24:31 +01:00
parent 61f6b94575
commit 2d72f7be03
+2 -6
View File
@@ -8,7 +8,7 @@ data:
noise_keV: 0.13
noise_threshold: 0.0 ### set values below (noise * noise_threshold) to zero
num_aug_ops: 1
normalize: true
normalize: false
batch_size_train: 4096
batch_size_val: 1024
@@ -33,8 +33,4 @@ training:
loss:
type: "weightedL1"
alpha: 7.0
beta: 3.0
logging:
save_dir: "Results"
model_dir: "Models"
beta: 3.0