Fix the loss calculation in 3-ph training

Configs/train_3photon.yaml

@@ -7,9 +7,9 @@ data:
   energy: 12 ### in keV
 
   batch_size_train: 4096
-  batch_size_val: 1024
-  batch_size_test: 1024
-  num_workers: 32
+  batch_size_val: 8192
+  batch_size_test: 8192
+  num_workers: 16
   train_file_range: [0, 12]
   val_file_range: [13, 14]
   test_file_range: [15, 15]
@@ -17,15 +17,16 @@ data:
   n_size: 9 ### size of sub-images containing 3 photons
 
 model:
-  version: "260529" 
+  version: "260611" # 260529
 
 training:
   epochs: 1000
   learning_rate: 1.0e-3
-  weight_decay: 1.0e-4
+  weight_decay: 0
   scheduler_factor: 0.7
   scheduler_patience: 5
-  checkpoint_epochs: [10, 30, 50, 100, 150, 300, 500, 1000]
+  checkpoint_epochs: [10, 20, 50, 100, 150, 300, 500, 1000]
 
 loss:
-  type: "three_point_set_loss_smooth_l1"
+  type: "three_point_set_loss_smooth_l1"
+  smooth_l1_beta: 0.01