update data_loader by adding self.args.rag

Author: nbbb24

ecg_bench/scripts/org_results.sh

@@ -9,7 +9,10 @@ data=("ecg_instruct_45k_mapped_1250")
 # normalized_rag_features=True
 
 checkpoints=(
-    'llama-3.2-3b-instruct_2_1_1024_0.0001_0.9_0.99_1e-08_500_0.01_True_None_None_None_True_combined_combined_1_system_prompt_True_False'
+    'llama-3.2-3b-instruct_2_1_1024_0.0001_0.9_0.99_1e-08_500_0.01_True_None_None_None_True_feature_combined_1_system_prompt_True_False'
+    'llama-3.2-3b-instruct_2_1_1024_0.0001_0.9_0.99_1e-08_500_0.01_True_None_None_None_True_feature_combined_1_system_prompt_None_False'
+    'llama-3.2-3b-instruct_2_1_1024_0.0001_0.9_0.99_1e-08_500_0.01_True_None_None_None_True_signal_combined_1_system_prompt_True_False'
+    'llama-3.2-3b-instruct_2_1_1024_0.0001_0.9_0.99_1e-08_500_0.01_True_None_None_None_True_signal_combined_1_system_prompt_None_False'
 )
 
 for d in "${data[@]}"; do

ecg_bench/utils/data_loader_utils.py

@@ -107,7 +107,7 @@ def setup_conversation_template(self, signal = None):
         elif 'gemma' in self.args.model:
             conv = get_conv_template('gemma')
         feature=None
-        if self.args.retrieval_base in ['feature', 'combined']:
+        if self.args.rag and self.args.retrieval_base in ['feature', 'combined']:
             original_feature=self.rag_db.feature_extractor.extract_rag_features(signal)
             feature=original_feature
             if self.args.normalized_rag_feature:
@@ -146,7 +146,7 @@ def append_messages_to_conv(self, conv, altered_text, signal=None):
             message_value = message_value.replace('<image>', '')
             message_value = message_value.replace('<ecg>', '')
             message_value = message_value.replace('image', 'signal').replace('Image', 'Signal')
-            if self.args.retrieval_base in ['feature', 'combined'] or self.args.retrieved_information in ['feature','combined']:
+            if self.args.rag and (self.args.retrieval_base in ['feature', 'combined'] or self.args.retrieved_information in ['feature','combined']):
                 original_feature=self.rag_db.feature_extractor.extract_rag_features(signal)
                 feature=original_feature
                 if self.args.normalized_rag_feature: