Как использовать run_classifer.py, пример реализации Bert в Pytorch для классификации Task? - PullRequest
Новая
       49

Как использовать run_classifer.py, пример реализации Bert в Pytorch для классификации Task?

1 голос
/ 15 мая 2019

Как использовать тонко настроенную модель Бёрта Пайторха для задачи классификации (CoLa)?

Я не вижу аргумента --do_predict в /examples/run_classifier.py.

Однако, --do_predict существует в исходной реализации Bert.

Точно настроенная модель получает сохранение в BERT_OUTPUT_DIR как pytorch_model.bin, но есть ли простой способ повторно использовать ее через командную строку?

Использование реализации Pytorch из: https://github.com/huggingface/pytorch-pretrained-BERT

Команда, которую я использую для выполнения кода: 

python run_classifier.py \
  --task_name CoLA \
  --do_train \
  --do_eval \
  --do_lower_case \
  --data_dir ./split/ \
  --bert_model bert-base-uncased \
  --max_seq_length 128 \
  --train_batch_size 32 \
  --learning_rate 2e-5 \
  --num_train_epochs 3.0 \
...