diff --git a/examples/run_lm_finetuning.py b/examples/lm_finetuning/simple_lm_finetuning.py similarity index 99% rename from examples/run_lm_finetuning.py rename to examples/lm_finetuning/simple_lm_finetuning.py index dd56c1ab..3eea4e7a 100644 --- a/examples/run_lm_finetuning.py +++ b/examples/lm_finetuning/simple_lm_finetuning.py @@ -33,9 +33,6 @@ from pytorch_pretrained_bert.modeling import BertForPreTraining from pytorch_pretrained_bert.tokenization import BertTokenizer from pytorch_pretrained_bert.optimization import BertAdam, warmup_linear -from torch.utils.data import Dataset -import random - logging.basicConfig(format='%(asctime)s - %(levelname)s - %(name)s - %(message)s', datefmt='%m/%d/%Y %H:%M:%S', level=logging.INFO)