limit annotation for testing purposes

saleemhamo · Jul 18, 2024 · 922f47d · 922f47d
1 parent 4901544
commit 922f47d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/.github/workflows/docker-image.yml b/.github/workflows/docker-image.yml
@@ -27,6 +27,6 @@ jobs:
         with:
           context: .
           push: true
-          tags: saleemhamo/ctfg_pipeline_coarse_grained:v0.17
+          tags: saleemhamo/ctfg_pipeline_coarse_grained:v0.18
 
           build-args: MOUNTED_CLAIM_DIRECTORY=/mnt/data
diff --git a/models/coarse_grained/train.py b/models/coarse_grained/train.py
@@ -52,6 +52,7 @@ def main():
         test_file=CHARADES_ANNOTATIONS_TEST
     )
     annotations = charades_sta.get_train_data()
+    annotations = annotations[:5]  # For testing purposes
 
     # Load feature extractor
     logger.info("Loading feature extractor.")