added command line flag for getting trigger patches from file

set default back to simulation
ctlearn-project · Jul 2, 2024 · d4edbdc · d4edbdc
1 parent 95f97e3
commit d4edbdc
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/ctlearn/default_config_files/triggerSingleCNN.yml b/ctlearn/default_config_files/triggerSingleCNN.yml
@@ -4,7 +4,7 @@ Data:
         number_of_trigger_patches: [7, 7]
         reco_cherenkov_photons: False
         include_nsb_patches: 'off'
-        get_trigger_patch: 'file'
+        get_trigger_patch: 'simulation'
     waveform_settings:
         waveform_type: 'raw'
         waveform_sequence_length: 5

diff --git a/ctlearn/default_config_files/triggerTRN.yml b/ctlearn/default_config_files/triggerTRN.yml
@@ -4,7 +4,7 @@ Data:
         number_of_trigger_patches: [7, 7]
         reco_cherenkov_photons: False
         include_nsb_patches: 'off'
-        get_trigger_patch: 'file'
+        get_trigger_patch: 'simulation'
     waveform_settings:
         waveform_type: 'raw'
         waveform_sequence_length: 5

diff --git a/ctlearn/run_model.py b/ctlearn/run_model.py
@@ -485,6 +485,12 @@ def main():
         action=argparse.BooleanOptionalAction,
         help="Flag, if the network should include NSB trigger patches at training phase. In prediction mode, the network should only predict on NSB trigger patches",
     )
+    parser.add_argument(
+        "--trigger_patches_from_file",
+        default=False,
+        action=argparse.BooleanOptionalAction,
+        help="Flag, if the trigger patches should be retrieved from an external file",
+    )
     parser.add_argument(
         "--pretrained_weights", "-w", help="Path to the pretrained weights"
     )
@@ -668,6 +674,8 @@ def main():
                 config["Data"]["trigger_settings"]["include_nsb_patches"] = "auto"
             else:
                 config["Data"]["trigger_settings"]["include_nsb_patches"] = "off"
+            if args.trigger_patches_from_file:
+                config["Data"]["trigger_settings"]["get_trigger_patch"] = "file"
 
         run_model(config, mode="train", debug=args.debug, log_to_file=args.log_to_file)
 
@@ -705,6 +713,8 @@ def main():
                                 config["Data"]["trigger_settings"]["include_nsb_patches"] = "all"
                             else:
                                 config["Data"]["trigger_settings"]["include_nsb_patches"] = "off"
+                            if args.trigger_patches_from_file:
+                                config["Data"]["trigger_settings"]["get_trigger_patch"] = "file"
                         if args.tel_types:
                             config["Data"]["selected_telescope_types"] = args.tel_types
                         if args.allowed_tels:
@@ -780,6 +790,8 @@ def main():
                         config["Data"]["trigger_settings"]["include_nsb_patches"] = "all"
                     else:
                         config["Data"]["trigger_settings"]["include_nsb_patches"] = "off"
+                    if args.trigger_patches_from_file:
+                        config["Data"]["trigger_settings"]["get_trigger_patch"] = "file"
                 if args.tel_types:
                     config["Data"]["selected_telescope_types"] = args.tel_types
                 if args.allowed_tels: