Skip to content
Permalink
Browse files

only download resnet50 and en2gr for expensive tests (#3030)

Summary:
We're downloading all models for RELEASE_WITH_EXPENSIVE_TESTS CI build, but looks like we only use resnet and en2gr text translation models.

Documentation:
Pull Request resolved: #3030

Differential Revision: D15588077

Pulled By: nickgg

fbshipit-source-id: 24140cb49d9c83d5f4f40d16e1bb86047956f3d7
  • Loading branch information...
nickgg authored and facebook-github-bot committed May 31, 2019
1 parent c7a9022 commit c9caa69148a1862f9ff0fea578656a52a41d4317
Showing with 2 additions and 1 deletion.
  1. +1 −1 .circleci/build.sh
  2. +1 −0 utils/download_datasets_and_models.py
@@ -67,7 +67,7 @@ elif [[ "${CIRCLE_JOB}" == "TSAN" ]]; then
elif [[ "$CIRCLE_JOB" == "RELEASE_WITH_EXPENSIVE_TESTS" ]]; then
# Download the models and tell cmake where to find them.
MODELS_DIR="$GLOW_DIR/downloaded_models"
DOWNLOAD_EXE="python $GLOW_DIR/utils/download_datasets_and_models.py --all-caffe2-models"
DOWNLOAD_EXE="python $GLOW_DIR/utils/download_datasets_and_models.py -c resnet50 en2gr"
mkdir $MODELS_DIR
(
cd $MODELS_DIR
@@ -162,6 +162,7 @@ def download(path, filename, url):

def download_caffe2_models(outDir, models):
for modelname in models:
print("For model ", modelname);
for filename in ["predict_net.pbtxt", "predict_net.pb", "init_net.pb"]:
path = os.path.join(outDir, modelname)
url = "http://fb-glow-assets.s3.amazonaws.com/models/{}/{}".format(

0 comments on commit c9caa69

Please sign in to comment.
You can’t perform that action at this time.