From 44f9c8b493b42b0a0ecad2f16804e8684dc0821d Mon Sep 17 00:00:00 2001 From: Xingjian Shi Date: Mon, 10 Aug 2020 05:07:45 -0700 Subject: [PATCH] Update benchmark_gluonnlp.py --- scripts/benchmarks/benchmark_gluonnlp.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/scripts/benchmarks/benchmark_gluonnlp.py b/scripts/benchmarks/benchmark_gluonnlp.py index 7d4c59b446..90a9537836 100644 --- a/scripts/benchmarks/benchmark_gluonnlp.py +++ b/scripts/benchmarks/benchmark_gluonnlp.py @@ -4,18 +4,18 @@ MODELS = [ - # 'google_en_uncased_bert_base', - # 'google_en_uncased_bert_large', - # 'google_albert_base_v2', - # 'google_albert_large_v2', - # 'google_albert_xlarge_v2', - # 'google_albert_xxlarge_v2', + 'google_en_uncased_bert_base', + 'google_en_uncased_bert_large', + 'google_albert_base_v2', + 'google_albert_large_v2', + 'google_albert_xlarge_v2', + 'google_albert_xxlarge_v2', 'google_electra_small', - # 'google_electra_base', - # 'google_electra_large', - # 'google_uncased_mobilebert', + 'google_electra_base', + 'google_electra_large', + 'google_uncased_mobilebert', 'fairseq_bart_base', - # 'fairseq_bart_large' + 'fairseq_bart_large' ] # (batch_size, seq_length)