HACK (индексы меток меняются в предварительно обученной модели RoBERTa) в run_glue.py - PullRequest
Новая
       21

HACK (индексы меток меняются в предварительно обученной модели RoBERTa) в run_glue.py

0 голосов
/ 10 апреля 2020

Внутри функции load_cache_examples in run_glue.py: 

if task in ["mnli", "mnli-mm"] and args.model_type in ["roberta", "xlmroberta"]:
            # HACK(label indices are swapped in RoBERTa pretrained model)
            label_list[1], label_list[2] = label_list[2], label_list[1]

Какова цель обмена индексами меток для набора данных MNLI? Нужно ли нам делать то же самое, если мы настраиваем другие наборы данных с 3 метками?

...