get_rescale_baseline

Computing rescale baselines for English models

Downloading the dataset

This downloads the WMT17 English text data.

bash download_text_data.sh

Tuning the models

Here is an example of getting the rescale baseline files of two models

python get_rescale_baseline.py --lang en -b 16 -m \
    microsoft/deberta-large \
    microsoft/deberta-large-mnli

The baseline files will be in rescale_baseline folder