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_swag.py
+++ b/examples/run_swag.py
@@ -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):