Update README.md
#3
by
astachowicz
- opened
README.md
CHANGED
@@ -24,7 +24,7 @@ The only difference is that there are a few new training arguments specific to H
|
|
24 |
|
25 |
[Here](https://github.com/huggingface/optimum-habana/blob/main/examples/question-answering/run_qa.py) is a question-answering example script to fine-tune a model on SQuAD. You can run it with RoBERTa Large with the following command:
|
26 |
```bash
|
27 |
-
python run_qa.py \
|
28 |
--model_name_or_path roberta-large \
|
29 |
--gaudi_config_name Habana/roberta-large \
|
30 |
--dataset_name squad \
|
@@ -37,7 +37,9 @@ python run_qa.py \
|
|
37 |
--max_seq_length 384 \
|
38 |
--output_dir /tmp/squad/ \
|
39 |
--use_habana \
|
40 |
-
--
|
|
|
|
|
41 |
--throughput_warmup_steps 3 \
|
42 |
--bf16
|
43 |
```
|
|
|
24 |
|
25 |
[Here](https://github.com/huggingface/optimum-habana/blob/main/examples/question-answering/run_qa.py) is a question-answering example script to fine-tune a model on SQuAD. You can run it with RoBERTa Large with the following command:
|
26 |
```bash
|
27 |
+
PT_HPU_LAZY_MODE=0 python run_qa.py \
|
28 |
--model_name_or_path roberta-large \
|
29 |
--gaudi_config_name Habana/roberta-large \
|
30 |
--dataset_name squad \
|
|
|
37 |
--max_seq_length 384 \
|
38 |
--output_dir /tmp/squad/ \
|
39 |
--use_habana \
|
40 |
+
--torch_compile_backend hpu_backend \
|
41 |
+
--torch_compile \
|
42 |
+
--use_lazy_mode false \
|
43 |
--throughput_warmup_steps 3 \
|
44 |
--bf16
|
45 |
```
|