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

fix some errors for distributed lm_finetuning

Browse Source
This commit is contained in:
yzy5630
2019-07-17 09:16:20 +08:00
parent 5fe0b378d8
commit 60a1bdcdac
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
examples/lm_finetuning/simple_lm_finetuning.py
View File

@@ -504,7 +504,7 @@ def main():
if os.path.exists(args.output_dir) and os.listdir(args.output_dir): if os.path.exists(args.output_dir) and os.listdir(args.output_dir):
raise ValueError("Output directory ({}) already exists and is not empty.".format(args.output_dir)) raise ValueError("Output directory ({}) already exists and is not empty.".format(args.output_dir))
if not os.path.exists(args.output_dir): if not os.path.exists(args.output_dir) and torch.distributed.get_rank() == 0:
os.makedirs(args.output_dir) os.makedirs(args.output_dir)
tokenizer = BertTokenizer.from_pretrained(args.bert_model, do_lower_case=args.do_lower_case) tokenizer = BertTokenizer.from_pretrained(args.bert_model, do_lower_case=args.do_lower_case)
@@ -613,11 +613,11 @@ def main():
global_step += 1 global_step += 1
# Save a trained model # Save a trained model
logger.info("** ** * Saving fine - tuned model ** ** * ")
model_to_save = model.module if hasattr(model, 'module') else model # Only save the model it-self model_to_save = model.module if hasattr(model, 'module') else model # Only save the model it-self
output_model_file = os.path.join(args.output_dir, WEIGHTS_NAME) output_model_file = os.path.join(args.output_dir, WEIGHTS_NAME)
output_config_file = os.path.join(args.output_dir, CONFIG_NAME) output_config_file = os.path.join(args.output_dir, CONFIG_NAME)
if args.do_train: if args.do_train and torch.distributed.get_rank() == 0:
logger.info("** ** * Saving fine - tuned model ** ** * ")
torch.save(model_to_save.state_dict(), output_model_file) torch.save(model_to_save.state_dict(), output_model_file)
model_to_save.config.to_json_file(output_config_file) model_to_save.config.to_json_file(output_config_file)
tokenizer.save_vocabulary(args.output_dir) tokenizer.save_vocabulary(args.output_dir)