提交 a50b2eeb 编写于 作者: L liuqi

Remove unsed code.

上级 33d4454d
......@@ -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
......
......@@ -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"
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册