diff --git a/examples/run_bert_classifier.py b/examples/run_bert_classifier.py index 8bb37159d30..d987b353216 100644 --- a/examples/run_bert_classifier.py +++ b/examples/run_bert_classifier.py @@ -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 diff --git a/examples/run_bert_squad.py b/examples/run_bert_squad.py index 9aaa711c2b7..54eceb36f74 100644 --- a/examples/run_bert_squad.py +++ b/examples/run_bert_squad.py @@ -33,7 +33,7 @@ from tqdm import tqdm, trange 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 BertForQuestionAnswering from pytorch_pretrained_bert.optimization import BertAdam, WarmupLinearSchedule from pytorch_pretrained_bert.tokenization import BertTokenizer diff --git a/examples/run_xlnet_classifier.py b/examples/run_xlnet_classifier.py index 23098159818..fb5501e3707 100644 --- a/examples/run_xlnet_classifier.py +++ b/examples/run_xlnet_classifier.py @@ -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_xlnet import XLNetForSequenceClassification from pytorch_pretrained_bert.tokenization_xlnet import XLNetTokenizer from pytorch_pretrained_bert.optimization import BertAdam, WarmupLinearSchedule diff --git a/examples/run_xlnet_squad.py b/examples/run_xlnet_squad.py index 927668c57a8..c299358b79b 100644 --- a/examples/run_xlnet_squad.py +++ b/examples/run_xlnet_squad.py @@ -33,7 +33,7 @@ from tqdm import tqdm, trange 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_xlnet import BertForQuestionAnswering from pytorch_pretrained_bert.tokenization_xlnet import XLNetTokenizer from pytorch_pretrained_bert.optimization import BertAdam, WarmupLinearSchedule