Chua, Vui Seng commited on
Commit
73501a1
1 Parent(s): 2524d65

Update eval batch size

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -105,9 +105,9 @@ nohup python run_qa.py \
105
  --nncf_config $MODELROOT/nncf_bert_squad_sparsity.json \
106
  --to_onnx $OUTDIR/bert-base-squadv1-block-pruning-hybrid-filled-lt-nncf-50.0sparse-qat-lt.onnx \
107
  --do_eval \
108
- --per_device_eval_batch_size 16 \
109
  --max_seq_length 384 \
110
  --doc_stride 128 \
111
  --overwrite_output_dir \
112
  --output_dir $OUTDIR 2>&1 | tee $OUTDIR/run.log &
113
- ```
 
105
  --nncf_config $MODELROOT/nncf_bert_squad_sparsity.json \
106
  --to_onnx $OUTDIR/bert-base-squadv1-block-pruning-hybrid-filled-lt-nncf-50.0sparse-qat-lt.onnx \
107
  --do_eval \
108
+ --per_device_eval_batch_size 128 \
109
  --max_seq_length 384 \
110
  --doc_stride 128 \
111
  --overwrite_output_dir \
112
  --output_dir $OUTDIR 2>&1 | tee $OUTDIR/run.log &
113
+ ```