Update README.md
Browse files
README.md
CHANGED
|
@@ -12,7 +12,7 @@ widget:
|
|
| 12 |
|
| 13 |
## DeBERTa: Decoding-enhanced BERT with Disentangled Attention
|
| 14 |
|
| 15 |
-
[DeBERTa](https://arxiv.org/abs/2006.03654) improves the BERT and RoBERTa models using disentangled attention and enhanced mask decoder.
|
| 16 |
|
| 17 |
Please check the [official repository](https://github.com/microsoft/DeBERTa) for more details and updates.
|
| 18 |
|
|
@@ -51,20 +51,20 @@ export TASK_NAME=rte
|
|
| 51 |
output_dir="ds_results"
|
| 52 |
num_gpus=8
|
| 53 |
batch_size=4
|
| 54 |
-
python -m torch.distributed.launch --nproc_per_node=${num_gpus} \
|
| 55 |
-
run_glue.py \
|
| 56 |
-
--model_name_or_path microsoft/deberta-v2-xxlarge-mnli \
|
| 57 |
-
--task_name $TASK_NAME \
|
| 58 |
-
--do_train \
|
| 59 |
-
--do_eval \
|
| 60 |
-
--max_seq_length 256 \
|
| 61 |
-
--per_device_train_batch_size ${batch_size} \
|
| 62 |
-
--learning_rate 3e-6 \
|
| 63 |
-
--num_train_epochs 3 \
|
| 64 |
-
--output_dir $output_dir \
|
| 65 |
-
--overwrite_output_dir \
|
| 66 |
-
--logging_steps 10 \
|
| 67 |
-
--logging_dir $output_dir \
|
| 68 |
--deepspeed ds_config.json
|
| 69 |
```
|
| 70 |
|
|
@@ -72,8 +72,8 @@ You can also run with `--sharded_ddp`
|
|
| 72 |
```bash
|
| 73 |
cd transformers/examples/text-classification/
|
| 74 |
export TASK_NAME=rte
|
| 75 |
-
python -m torch.distributed.launch --nproc_per_node=8 run_glue.py --model_name_or_path microsoft/deberta-v2-xxlarge-mnli \
|
| 76 |
-
--task_name $TASK_NAME --do_train --do_eval --max_seq_length 256 --per_device_train_batch_size 4 \
|
| 77 |
--learning_rate 3e-6 --num_train_epochs 3 --output_dir /tmp/$TASK_NAME/ --overwrite_output_dir --sharded_ddp --fp16
|
| 78 |
```
|
| 79 |
|
|
|
|
| 12 |
|
| 13 |
## DeBERTa: Decoding-enhanced BERT with Disentangled Attention
|
| 14 |
|
| 15 |
+
[DeBERTa](https://arxiv.org/abs/2006.03654) improves the BERT and RoBERTa models using disentangled attention and enhanced mask decoder. It outperforms BERT and RoBERTa on majority of NLU tasks with 80GB training data.
|
| 16 |
|
| 17 |
Please check the [official repository](https://github.com/microsoft/DeBERTa) for more details and updates.
|
| 18 |
|
|
|
|
| 51 |
output_dir="ds_results"
|
| 52 |
num_gpus=8
|
| 53 |
batch_size=4
|
| 54 |
+
python -m torch.distributed.launch --nproc_per_node=${num_gpus} \\
|
| 55 |
+
run_glue.py \\
|
| 56 |
+
--model_name_or_path microsoft/deberta-v2-xxlarge-mnli \\
|
| 57 |
+
--task_name $TASK_NAME \\
|
| 58 |
+
--do_train \\
|
| 59 |
+
--do_eval \\
|
| 60 |
+
--max_seq_length 256 \\
|
| 61 |
+
--per_device_train_batch_size ${batch_size} \\
|
| 62 |
+
--learning_rate 3e-6 \\
|
| 63 |
+
--num_train_epochs 3 \\
|
| 64 |
+
--output_dir $output_dir \\
|
| 65 |
+
--overwrite_output_dir \\
|
| 66 |
+
--logging_steps 10 \\
|
| 67 |
+
--logging_dir $output_dir \\
|
| 68 |
--deepspeed ds_config.json
|
| 69 |
```
|
| 70 |
|
|
|
|
| 72 |
```bash
|
| 73 |
cd transformers/examples/text-classification/
|
| 74 |
export TASK_NAME=rte
|
| 75 |
+
python -m torch.distributed.launch --nproc_per_node=8 run_glue.py --model_name_or_path microsoft/deberta-v2-xxlarge-mnli \\
|
| 76 |
+
--task_name $TASK_NAME --do_train --do_eval --max_seq_length 256 --per_device_train_batch_size 4 \\
|
| 77 |
--learning_rate 3e-6 --num_train_epochs 3 --output_dir /tmp/$TASK_NAME/ --overwrite_output_dir --sharded_ddp --fp16
|
| 78 |
```
|
| 79 |
|