File tree 1 file changed +4
-4
lines changed
1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change 17
17
# Prepare dataset
18
18
git clone https://github.com/intel/neural-compressor.git
19
19
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
21
21
22
22
# 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
24
24
25
25
# 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
28
28
29
29
# Run quantization using INC
30
30
bash run_quant.sh --input_model=./bert_fp32.pb --output_model=./bert_int8.pb --dataset_location=./data
You can’t perform that action at this time.
0 commit comments