Chua, Vui Seng
commited on
Commit
•
2524d65
1
Parent(s):
fae2108
Update training snippet in readme
Browse files
README.md
CHANGED
@@ -41,19 +41,20 @@ pip install onnx
|
|
41 |
git clone https://huggingface.co/vuiseng9/bert-base-squadv1-block-pruning-hybrid-filled-lt
|
42 |
BASE_MODEL=/path/to/cloned_repo_above #to-revise
|
43 |
|
44 |
-
wget https://huggingface.co/vuiseng9/bert-base-squadv1-block-pruning-hybrid-filled-lt-nncf-50.0sparse-qat-lt/nncf_bert_squad_sparsity.json
|
45 |
NNCF_CFG=/path/to/downloaded_nncf_cfg_above #to-revise
|
46 |
|
47 |
-
export CUDA_VISIBLE_DEVICES=0
|
48 |
-
NEPOCH=5
|
49 |
-
|
50 |
-
RUNID=bert-base-squadv1-block-pruning-hybrid-filled-lt-nncf-50.0sparse-qat-lt
|
51 |
OUTROOT=/path/to/train_output_root #to-revise
|
52 |
WORKDIR=transformers/examples/pytorch/question-answering #to-revise
|
|
|
|
|
|
|
53 |
|
54 |
OUTDIR=$OUTROOT/$RUNID
|
55 |
mkdir -p $OUTDIR
|
56 |
-
|
|
|
|
|
57 |
|
58 |
python run_qa.py \
|
59 |
--model_name_or_path vuiseng9/bert-base-squadv1-block-pruning-hybrid \
|
|
|
41 |
git clone https://huggingface.co/vuiseng9/bert-base-squadv1-block-pruning-hybrid-filled-lt
|
42 |
BASE_MODEL=/path/to/cloned_repo_above #to-revise
|
43 |
|
44 |
+
wget https://huggingface.co/vuiseng9/bert-base-squadv1-block-pruning-hybrid-filled-lt-nncf-50.0sparse-qat-lt/raw/main/nncf_bert_squad_sparsity.json
|
45 |
NNCF_CFG=/path/to/downloaded_nncf_cfg_above #to-revise
|
46 |
|
|
|
|
|
|
|
|
|
47 |
OUTROOT=/path/to/train_output_root #to-revise
|
48 |
WORKDIR=transformers/examples/pytorch/question-answering #to-revise
|
49 |
+
RUNID=bert-base-squadv1-block-pruning-hybrid-filled-lt-nncf-50.0sparse-qat-lt
|
50 |
+
|
51 |
+
cd $WORKDIR
|
52 |
|
53 |
OUTDIR=$OUTROOT/$RUNID
|
54 |
mkdir -p $OUTDIR
|
55 |
+
|
56 |
+
export CUDA_VISIBLE_DEVICES=0
|
57 |
+
NEPOCH=5
|
58 |
|
59 |
python run_qa.py \
|
60 |
--model_name_or_path vuiseng9/bert-base-squadv1-block-pruning-hybrid \
|