Update loss fns

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
+  type: "huber_loss" 
+  huber_beta: 0.01

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"
+  type: "two_point_set_loss_smooth_l1" #"two_point_set_loss_l2"