changed default function back to main for generate_description

Author: Ye Shaokai

llava/action/generate_description.py

@@ -26,7 +26,7 @@ def generate_train_ann(ann_file, labels, mapping_vn2narration, mapping_vn2act, v
         mc_generator = AvionMultiChoiceGenerator(ann_root)
         with open(prediction_path, 'r') as f:
             train_predictions = json.load(f)         
-
+    print ('train_predictions', train_predictions)
     import spacy
     nlp = spacy.load('en_core_web_sm')
 
@@ -300,9 +300,9 @@ def fix_annotations():
 
 if __name__ == "__main__":
 
-    #main()
+    main()
 
-    fix_annotations()
+    #fix_annotations()
 
     # reason_path = "/storage-rcp-pure/upmwmathis_scratch/shaokai/train_anno_gpt-gt-reason_4_all.jsonl"
     # mc_path = "/storage-rcp-pure/upmwmathis_scratch/shaokai/EK100_inst_train/avion_mc_top5_GT_random_narration/train_convs_narration.jsonl"

shaokai_generate_train.sh

@@ -6,10 +6,10 @@ export PYTHONPATH=/usr/local/lib/python3.10/site-packages/decord-0.6.0-py3.10-li
 python3 llava/action/generate_description.py \
     --train_metadata /storage-rcp-pure/upmwmathis_scratch/shaokai/epic-kitchens-100-annotations/EPIC_100_train.csv \
     --out_folder /storage-rcp-pure/upmwmathis_scratch/shaokai/EK100_inst_train/ \
-    --train_predictions /storage-rcp-pure/upmwmathis_scratch/shaokai/TIM_PREDS/tim_pred_ids_train.json \
-    --gen_type tim_mc \
+    --train_predictions /storage-rcp-pure/upmwmathis_scratch/shaokai/TIM_PREDS/avion_pred_ids_train.json \
+    --gen_type avion_mc \
     --action_representation official_key \
-    --n_options 5
+    --n_options 20
 
 
 # python3 action/generate_description.py \