wehub-resource-sync
53 lines
2.3 KiB
Bash
53 lines
2.3 KiB
Bash
#!/bin/bash
|
|
# Copyright 2022 The TensorFlow Authors. All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
# ==============================================================================
|
|
|
|
source onednn_benchmark_config.sh
|
|
|
|
# Store models in home directory
|
|
mkdir -p ${TF_GRAPHS}
|
|
cd ${TF_GRAPHS}
|
|
|
|
# Download TF graphs linked from MLPerf Inference v2.0
|
|
# https://github.com/mlcommons/inference#mlperf-inference-v20-submission-02252022
|
|
wget -c https://zenodo.org/record/2535873/files/resnet50_v1.pb -O resnet50_v1-5.pb
|
|
wget -c https://storage.googleapis.com/download.tensorflow.org/models/inception5h.zip
|
|
wget -c https://zenodo.org/record/2269307/files/mobilenet_v1_1.0_224.tgz -O mobilenet-v1.tgz # 90MB
|
|
wget -c http://download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_coco_2018_01_28.tar.gz -O ssd-mobilenet-v1.tar.gz
|
|
wget -c https://zenodo.org/record/3345892/files/tf_ssd_resnet34_22.1.zip -O ssd-resnet34.zip
|
|
wget -c https://zenodo.org/record/3939747/files/model.pb -O bert-large.pb # 1.2GB. Could take 10+ minutes to download.
|
|
|
|
# Extract graphs from tar/zip files.
|
|
unzip ./inception5h.zip -d inception
|
|
mv inception/tensorflow_inception_graph.pb inception.pb
|
|
rm -rf inception
|
|
tar -xzf mobilenet-v1.tgz ./mobilenet_v1_1.0_224_frozen.pb
|
|
mv mobilenet_v1_1.0_224_frozen.pb mobilenet-v1.pb
|
|
tar -xzf ssd-mobilenet-v1.tar.gz ssd_mobilenet_v1_coco_2018_01_28/frozen_inference_graph.pb
|
|
mv ssd_mobilenet_v1_coco_2018_01_28/frozen_inference_graph.pb ssd-mobilenet-v1.pb
|
|
rm -rf ssd_mobilenet_v1_coco_2018_01_28
|
|
unzip ssd-resnet34.zip
|
|
mv tf_ssd_resnet34_22.1/resnet34_tf.22.1.pb ssd-resnet34.pb
|
|
rm -rf tf_ssd_resnet34_22.1
|
|
|
|
# Now we should have the following model files in ~/tf-graphs:
|
|
# - bert-large.pb
|
|
# - inception.pb
|
|
# - mobilenet-v1.pb
|
|
# - resnet50_v1-5.pb
|
|
# - ssd-mobilenet-v1.pb
|
|
# - ssd-resnet34.pb
|
|
|