detectron2: require coco2017-minimal data

It seems like detectron2 requires coco2017-minimal data. It gets installed with vision-maskrcnn, but if you want to install requirements only for one model we need the installer scripts for the detectron models to download coco2017-minimal.
pytorch · Nov 2, 2023 · d492158 · d492158
1 parent 94f54e8
commit d492158
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchbenchmark/util/framework/detectron2/__init__.py b/torchbenchmark/util/framework/detectron2/__init__.py
@@ -55,6 +55,7 @@ def remove_tools_directory():
 
 def install_detectron2(model_name, model_dir):
     s3_utils.checkout_s3_data("INPUT_TARBALLS", "coco128.tar.gz", decompress=True)
+    s3_utils.checkout_s3_data("INPUT_TARBALLS", "coco2017-minimal.tar.gz", decompress=True)
     install_model_weights(model_name, model_dir)
     pip_install_requirements()
     remove_tools_directory()