diff --git a/open_flamingo/eval/evaluate.py b/open_flamingo/eval/evaluate.py index 8f3632ec..8b420f72 100644 --- 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() \ No newline at end of file + main()