Skip to content

Commit cb94f08

Browse files
author
tylertitsworth
committed
censor downloads in tf INC tests
Signed-off-by: tylertitsworth <tyler.titsworth@intel.com>
1 parent 3643e7f commit cb94f08

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tensorflow/tests/inc_test.sh

+4-4
Original file line numberDiff line numberDiff line change
@@ -17,14 +17,14 @@
1717
# Prepare dataset
1818
git clone https://github.com/intel/neural-compressor.git
1919
cd neural-compressor/examples/tensorflow/nlp/bert_large_squad/quantization/ptq || exit
20-
bash prepare_dataset.sh --output_dir=./data
20+
bash prepare_dataset.sh --output_dir=./data >/dev/null 2>&1
2121

2222
# Preprocess the dataset
23-
python create_tf_record.py --vocab_file=./data/vocab.txt --predict_file=./data/dev-v1.1.json --output_file=./data/eval.tf_record
23+
python create_tf_record.py --vocab_file=./data/vocab.txt --predict_file=./data/dev-v1.1.json --output_file=./data/eval.tf_record >/dev/null 2>&1
2424

2525
# Download model
26-
bash prepare_model.sh --output_dir=./model
27-
python freeze_estimator_to_pb.py --input_model=./model --output_model=./bert_fp32.pb
26+
bash prepare_model.sh --output_dir=./model >/dev/null 2>&1
27+
python freeze_estimator_to_pb.py --input_model=./model --output_model=./bert_fp32.pb >/dev/null 2>&1
2828

2929
#Run quantization using INC
3030
bash run_quant.sh --input_model=./bert_fp32.pb --output_model=./bert_int8.pb --dataset_location=./data

0 commit comments

Comments
 (0)