From a50b2eeb2a517e99123eee9353d5506d00678e4e Mon Sep 17 00:00:00 2001 From: liuqi Date: Tue, 5 Dec 2017 17:41:23 +0800 Subject: [PATCH] Remove unsed code. --- mace/examples/mace_run.cc | 2 -- tools/validate_gcn.sh | 26 +++++++++++++------------- 2 files changed, 13 insertions(+), 15 deletions(-) diff --git a/mace/examples/mace_run.cc b/mace/examples/mace_run.cc index 0905477e..21eade9c 100644 --- a/mace/examples/mace_run.cc +++ b/mace/examples/mace_run.cc @@ -100,10 +100,8 @@ int main(int argc, char **argv) { in_file.close(); } - VLOG(0) << "Before Init"; // Init model auto net = CreateNet(net_def, &ws, device_type, NetMode::INIT); - VLOG(0) << "Before Run"; net->Run(); // run model diff --git a/tools/validate_gcn.sh b/tools/validate_gcn.sh index 279f1165..a1174466 100644 --- a/tools/validate_gcn.sh +++ b/tools/validate_gcn.sh @@ -20,19 +20,19 @@ PHONE_DATA_DIR="/data/local/tmp/${MACE_MODEL_NAME}" KERNEL_DIR="${PHONE_DATA_DIR}/cl/" # Step 1: convert tf model to mace model -#echo "Step 1: convert tf model to mace model" -#bazel build //mace/python/tools:tf_converter -#bazel-bin/mace/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \ -# --output=${MODEL_DIR}/${MACE_MODEL_NAME} \ -# --input_node=input \ -# --output_node=GCN/br_result_2/fcn_br \ -# --runtime=gpu -# -## Step 2: Generate input data -#echo "Step 2: Generate input data" -#python tools/validate.py --generate_data true --random_seed 1 \ -# --input_file=${MODEL_DIR}/${INPUT_FILE_NAME} \ -# --input_shape=512,512,3 +echo "Step 1: convert tf model to mace model" +bazel build //mace/python/tools:tf_converter +bazel-bin/mace/python/tools/tf_converter --input=${TF_MODEL_FILE_PATH} \ + --output=${MODEL_DIR}/${MACE_MODEL_NAME} \ + --input_node=input \ + --output_node=GCN/br_result_2/fcn_br \ + --runtime=gpu + +# Step 2: Generate input data +echo "Step 2: Generate input data" +python tools/validate.py --generate_data true --random_seed 1 \ + --input_file=${MODEL_DIR}/${INPUT_FILE_NAME} \ + --input_shape=512,512,3 # Step 3: Run model on the phone echo "Step 3: Run model on the phone" -- GitLab