Use Filelock to ensure distributed barriers

see context in https://github.com/huggingface/transformers/pull/4223
This commit is contained in:
Julien Chaumond committed 2020-05-14 11:58:32 -04:00
1 parent 015f7812ed
commit c547f15a17
6 files changed
+25 -33

No files matched your search

@@ -118,13 +118,9 @@ class DataTrainingArguments:
def get_dataset(args: DataTrainingArguments, tokenizer: PreTrainedTokenizer, evaluate=False, local_rank=-1):
file_path = args.eval_data_file if evaluate else args.train_data_file
if args.line_by_line:
return LineByLineTextDataset(
tokenizer=tokenizer, file_path=file_path, block_size=args.block_size, local_rank=local_rank
)
return LineByLineTextDataset(tokenizer=tokenizer, file_path=file_path, block_size=args.block_size)
else:
return TextDataset(
tokenizer=tokenizer, file_path=file_path, block_size=args.block_size, local_rank=local_rank,
)
return TextDataset(tokenizer=tokenizer, file_path=file_path, block_size=args.block_size)
def main():