c-my/transformers
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix #1015

Browse Source
This commit is contained in:
Peng Qi committed 2019-08-19 13:07:00 -07:00
1 parent c589862b78
commit a368b87791
1 file changed
+1 -1
Download Patch File Download Diff File Expand all files Collapse all files
examples/run_squad.py
+1 -1
View File
@@ -498,7 +498,7 @@ def main():
# Load a trained model and vocabulary that you have fine-tuned
model = model_class.from_pretrained(args.output_dir)
tokenizer = tokenizer_class.from_pretrained(args.output_dir)
tokenizer = tokenizer_class.from_pretrained(args.output_dir, do_lower_case=args.do_lower_case)
model.to(args.device)