no fp16 on evaluation

2018-12-13 14:54:02 +01:00
parent 087798b7fa
commit e1eab59aac
3 changed files with 0 additions and 6 deletions
--- a/examples/run_classifier.py
+++ b/examples/run_classifier.py
@@ -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):