Introduction

August 14, 2021 ยท View on GitHub

This Repo conduct language understanding task.

Requirements and Installation

  • PyTorch version == 1.8
  • Python version >= 3.6

To install fairseq from source and develop locally:

git clone https://github.com/dropreg/R-Drop.git
cd R-Drop/huggingface_transformer/
pip install --editable .

Fine-tuning on GLUE task:

Example fine-tuning cmd for MRPC task

cd bert_rdrop/
bash run.sh

You can run it on Roberta by simply switching hyper-parameter: model_name_or_path

--model_name_or_path roberta-base \

For each of the GLUE task, you will need to use following cmd-line arguments:

ModelMNLIQNLIQQPRTESST-2MRPCCoLASTS-B
--lr1e-51e-51e-51e-51e-51e-51e-51e-5
--batch-size323232832161616
--total-num-update12387333112113272203620935229653363598
--warmup-updates74321986283181221256137320214