add coco dataset name to evaluate.py

mlfoundations · Oct 25, 2023 · 55a8f7c · 55a8f7c
1 parent e104a0a
commit 55a8f7c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/open_flamingo/eval/evaluate.py b/open_flamingo/eval/evaluate.py
@@ -534,7 +534,7 @@ def main():
     if args.eval_coco:
         eval_dataset(
             args,
-            dataset_name="flickr30",
+            dataset_name="coco",
             eval_model=eval_model,
             results=results,
             eval_fn=evaluate_captioning,
@@ -1302,4 +1302,4 @@ def evaluate_classification(
 
 
 if __name__ == "__main__":
-    main()
+    main()