SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

WIP XLM + refactoring

Browse Source
This commit is contained in:
thomwolf
2019-07-03 22:54:39 +02:00
parent 288be7b7ea
commit c41f2bad69
27 changed files with 515 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/run_bert_classifier.py
View File

@@ -35,8 +35,8 @@ from torch.nn import CrossEntropyLoss, MSELoss
from tensorboardX import SummaryWriter
from pytorch_pretrained_bert import WEIGHTS_NAME, CONFIG_NAME
from pytorch_pretrained_bert.modeling import BertForSequenceClassification
from pytorch_pretrained_bert.tokenization import BertTokenizer
from pytorch_pretrained_bert.modeling_bert import BertForSequenceClassification
from pytorch_pretrained_bert.tokenization_bert import BertTokenizer
from pytorch_pretrained_bert.optimization import BertAdam, WarmupLinearSchedule
from utils_glue import processors, output_modes, convert_examples_to_features, compute_metrics