Skip to content

Commit

Permalink
remove coco128
Browse files Browse the repository at this point in the history
  • Loading branch information
davidberard98 committed Nov 2, 2023
1 parent d492158 commit 035e296
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion torchbenchmark/util/framework/detectron2/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,6 @@ def remove_tools_directory():
pass

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()
Expand Down

0 comments on commit 035e296

Please sign in to comment.