fix imports

This commit is contained in:
thomwolf committed 2019-06-27 17:18:46 +02:00
1 parent d939d6fd02
commit 3a00674cbf
4 files changed
+4 -4

No files matched your search

+1 -1
View File
@@ -34,7 +34,7 @@ from torch.nn import CrossEntropyLoss, MSELoss
from tensorboardX import SummaryWriter
from pytorch_pretrained_bert.file_utils import WEIGHTS_NAME, CONFIG_NAME
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.optimization import BertAdam, WarmupLinearSchedule