From 6e2e47cfd617e764a615965654eb94b91ee86ca4 Mon Sep 17 00:00:00 2001
From: "xiangyu.xie" <xiangyu.xie@psi.ch>
Date: Fri, 29 May 2026 16:11:31 +0200
Subject: [PATCH] Update loss fns

---
 Configs/train_1photon.yaml | 13 ++++++-------
 Configs/train_2photon.yaml |  8 ++++----
 2 files changed, 10 insertions(+), 11 deletions(-)

diff --git a/Configs/train_1photon.yaml b/Configs/train_1photon.yaml
index 901b1c0..ea8b974 100644
--- a/Configs/train_1photon.yaml
+++ b/Configs/train_1photon.yaml
@@ -11,24 +11,23 @@ data:
   normalize: false
 
   batch_size_train: 4096
-  batch_size_val: 1024
-  batch_size_test: 1024
+  batch_size_val: 8192
+  batch_size_test: 8192
   num_workers: 32
   train_file_range: [0, 12]
   val_file_range: [13, 14]
   test_file_range: [15, 15]
 
 model:
-  version: "251022"
+  version: "260511" # 251022
 
 training:
   epochs: 150
   learning_rate: 1.0e-3
   scheduler_factor: 0.7
   scheduler_patience: 3
-  checkpoint_epochs: [10, 30, 50, 100, 150]
+  checkpoint_epochs: [10, 30, 50, 100, 150, 200, 300, 500, 1000]
 
 loss:
-  type: "weightedL1"
-  alpha: 7.0
-  beta: 3.0
\ No newline at end of file
+  type: "huber_loss" 
+  huber_beta: 0.01
\ No newline at end of file
diff --git a/Configs/train_2photon.yaml b/Configs/train_2photon.yaml
index fa3852e..9bc79e9 100644
--- a/Configs/train_2photon.yaml
+++ b/Configs/train_2photon.yaml
@@ -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"
\ No newline at end of file
+  type: "two_point_set_loss_smooth_l1" #"two_point_set_loss_l2"
\ No newline at end of file