Update loss fns

2026-05-29 16:11:31 +02:00
parent eeb507ec18
commit 6e2e47cfd6
2 changed files with 10 additions and 11 deletions
@@ -19,15 +19,15 @@ data:
  n_size: 7 ### size of sub-images containing 2 photons

 model:
-  version: "251124"
+  version: "260507" ## 251124

 training:
-  epochs: 150
+  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]
+  checkpoint_epochs: [10, 30, 50, 100, 150, 300, 500, 1000]

 loss:
-  type: "two_point_set_loss_l2"
+  type: "two_point_set_loss_smooth_l1" #"two_point_set_loss_l2"