From 76be189b08941adc13f07bdeb57e511b68b67290 Mon Sep 17 00:00:00 2001 From: Peiqin Lin Date: Sun, 21 Jul 2019 20:39:42 +0800 Subject: [PATCH] typos --- examples/run_glue.py | 4 ++-- examples/run_squad.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/run_glue.py b/examples/run_glue.py index f017db2f6f..25a487156e 100644 --- a/examples/run_glue.py +++ b/examples/run_glue.py @@ -116,8 +116,8 @@ def train(args, train_dataset, model, tokenizer): 'attention_mask': batch[1], 'token_type_ids': batch[2] if args.model_type in ['bert', 'xlnet'] else None, # XLM don't use segment_ids 'labels': batch[3]} - ouputs = model(**inputs) - loss = ouputs[0] # model outputs are always tuple in pytorch-transformers (see doc) + outputs = model(**inputs) + loss = outputs[0] # model outputs are always tuple in pytorch-transformers (see doc) if args.n_gpu > 1: loss = loss.mean() # mean() to average on multi-gpu parallel training diff --git a/examples/run_squad.py b/examples/run_squad.py index d72d67b87d..53ea0bfd64 100644 --- a/examples/run_squad.py +++ b/examples/run_squad.py @@ -129,8 +129,8 @@ def train(args, train_dataset, model, tokenizer): if args.model_type in ['xlnet', 'xlm']: inputs.update({'cls_index': batch[5], 'p_mask': batch[6]}) - ouputs = model(**inputs) - loss = ouputs[0] # model outputs are always tuple in pytorch-transformers (see doc) + outputs = model(**inputs) + loss = outputs[0] # model outputs are always tuple in pytorch-transformers (see doc) if args.n_gpu > 1: loss = loss.mean() # mean() to average on multi-gpu parallel (not distributed) training