Add missing arguemtn to run_clip.py (#21588)

2023-02-13 09:27:23 -06:00
parent 1210c72e82
commit fd5320bb57
1 changed files with 4 additions and 0 deletions
--- a/examples/pytorch/contrastive-image-text/run_clip.py
+++ b/examples/pytorch/contrastive-image-text/run_clip.py
@@ -131,6 +131,10 @@ class DataTrainingArguments:
        default=None,
        metadata={"help": "An optional input evaluation data file (a jsonlines file)."},
    )
+    test_file: Optional[str] = field(
+        default=None,
+        metadata={"help": "An optional input testing data file (a jsonlines file)."},
+    )
    max_seq_length: Optional[int] = field(
        default=128,
        metadata={