fix the evaluation on cluster

Author: HaozheQi

action/dataset.py

@@ -671,7 +671,6 @@ def get_topk_predictions(data, idx,  k):
 
 
 if __name__ == '__main__':
-    from moviepy.editor import ImageSequenceClip
     import torchvision
     import torchvision.transforms._transforms_video as transforms_video
 

run_EK100.sh

@@ -1,9 +1,9 @@
 python3 action/dataset.py \
-    --root /media/data/haozhe/VFM/EK100/EK100_320p_15sec_30fps_libx264 \
+    --root /media/data/haozhe/VFM/onevision/llava_video/EK100 \
     --train-metadata /media/data/haozhe/VFM/EK100/epic-kitchens-100-annotations/EPIC_100_train.csv \
     --val-metadata /media/data/haozhe/VFM/EK100/epic-kitchens-100-annotations/EPIC_100_validation.csv \
-    --llm_size 7b \
-    --llava_num_frames 16 > kitchen_test.out 2>&1 \
-    # --llava_checkpoint /data/epic_kitchen/EK100_test/checkpoint-8402						    
-    # --action_predictions action/avaion_predictions.json \
-    # --topk_predictions 10
+    --llm_size 0.5b \
+    --llava_num_frames 16 \
+    --llava_checkpoint experiments/EK100_test/checkpoint-8402 \
+    --action_predictions action/avaion_predictions.json \
+    --topk_predictions 10 > kitchen_test.out 2>&1