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

Update training snippet in readme

Browse files
Files changed (1) hide show
  1. README.md +7 -6
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
- cd $WORKDIR
 
 
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 \