Add inference codes for 3-ph category

Co-authored-by: Copilot <copilot@github.com>
2026-06-01 14:21:37 +02:00
parent 2e4f22d062
commit b3c08ad2e5
4 changed files with 285 additions and 2 deletions
@@ -13,14 +13,14 @@ data:
  train_file_range: [0, 12]
  val_file_range: [13, 14]
  test_file_range: [15, 15]
-  sample_ratio: 0.1
+  sample_ratio: 1.0
  n_size: 9 ### size of sub-images containing 3 photons

 model:
  version: "260529" 

 training:
-  epochs: 100
+  epochs: 1000
  learning_rate: 1.0e-3
  weight_decay: 1.0e-4
  scheduler_factor: 0.7