Skip to content

Commit 1e03c23

Browse files
davidberard98facebook-github-bot
authored andcommitted
detectron2: require coco2017-minimal data (#2022)
Summary: 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. Pull Request resolved: #2022 Reviewed By: xuzhao9 Differential Revision: D50915910 Pulled By: davidberard98 fbshipit-source-id: 54d9e7404dcb8ec2a277fd5dd03903f539a56a4b
1 parent 94f54e8 commit 1e03c23

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchbenchmark/util/framework/detectron2/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,7 @@ def remove_tools_directory():
5454
pass
5555

5656
def install_detectron2(model_name, model_dir):
57-
s3_utils.checkout_s3_data("INPUT_TARBALLS", "coco128.tar.gz", decompress=True)
57+
s3_utils.checkout_s3_data("INPUT_TARBALLS", "coco2017-minimal.tar.gz", decompress=True)
5858
install_model_weights(model_name, model_dir)
5959
pip_install_requirements()
6060
remove_tools_directory()

0 commit comments

Comments
 (0)