MLXID/DeepLearning
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update loss fns

Browse Source
This commit is contained in:
xie_x1
2026-05-29 16:11:31 +02:00
parent eeb507ec18
commit 6e2e47cfd6
2 changed files with 10 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Configs/train_1photon.yaml
+6 -7
View File
@@ -11,24 +11,23 @@ data:
normalize: false normalize: false
batch_size_train: 4096 batch_size_train: 4096
batch_size_val: 1024 batch_size_val: 8192
batch_size_test: 1024 batch_size_test: 8192
num_workers: 32 num_workers: 32
train_file_range: [0, 12] train_file_range: [0, 12]
val_file_range: [13, 14] val_file_range: [13, 14]
test_file_range: [15, 15] test_file_range: [15, 15]
model: model:
version: "251022" version: "260511" # 251022
training: training:
epochs: 150 epochs: 150
learning_rate: 1.0e-3 learning_rate: 1.0e-3
scheduler_factor: 0.7 scheduler_factor: 0.7
scheduler_patience: 3 scheduler_patience: 3
checkpoint_epochs: [10, 30, 50, 100, 150] checkpoint_epochs: [10, 30, 50, 100, 150, 200, 300, 500, 1000]
loss: loss:
type: "weightedL1" type: "huber_loss"
alpha: 7.0 huber_beta: 0.01
beta: 3.0
Configs/train_2photon.yaml
+4 -4
View File
@@ -19,15 +19,15 @@ data:
n_size: 7 ### size of sub-images containing 2 photons n_size: 7 ### size of sub-images containing 2 photons
model: model:
version: "251124" version: "260507" ## 251124
training: training:
epochs: 150 epochs: 1000
learning_rate: 1.0e-3 learning_rate: 1.0e-3
weight_decay: 1.0e-4 weight_decay: 1.0e-4
scheduler_factor: 0.7 scheduler_factor: 0.7
scheduler_patience: 5 scheduler_patience: 5
checkpoint_epochs: [10, 30, 50, 100, 150] checkpoint_epochs: [10, 30, 50, 100, 150, 300, 500, 1000]
loss: loss:
type: "two_point_set_loss_l2" type: "two_point_set_loss_smooth_l1" #"two_point_set_loss_l2"