riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 19:21:31 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

no fp16 on evaluation

Browse Source
This commit is contained in:
thomwolf 2018-12-13 14:54:02 +01:00
parent 087798b7fa
commit e1eab59aac
3 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/run_classifier.py
View File

@ -558,8 +558,6 @@ def main():
# Load a trained model that you have fine-tuned
model_state_dict = torch.load(output_model_file)
model = BertForSequenceClassification.from_pretrained(args.bert_model, state_dict=model_state_dict)
if args.fp16:
model.half()
model.to(device)
if args.do_eval and (args.local_rank == -1 or torch.distributed.get_rank() == 0):

2
examples/run_squad.py
View File

@ -923,8 +923,6 @@ def main():
# Load a trained model that you have fine-tuned
model_state_dict = torch.load(output_model_file)
model = BertForQuestionAnswering.from_pretrained(args.bert_model, state_dict=model_state_dict)
if args.fp16:
model.half()
model.to(device)
if args.do_predict and (args.local_rank == -1 or torch.distributed.get_rank() == 0):

2
examples/run_swag.py
View File

@ -478,8 +478,6 @@ def main():
model = BertForMultipleChoice.from_pretrained(args.bert_model,
state_dict=model_state_dict,
num_choices=4)
if args.fp16:
model.half()
model.to(device)
if args.do_eval and (args.local_rank == -1 or torch.distributed.get_rank() == 0):