We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 45c58b6 commit 44f9c8bCopy full SHA for 44f9c8b
1 file changed
scripts/benchmarks/benchmark_gluonnlp.py
@@ -4,18 +4,18 @@
4
5
6
MODELS = [
7
- # 'google_en_uncased_bert_base',
8
- # 'google_en_uncased_bert_large',
9
- # 'google_albert_base_v2',
10
- # 'google_albert_large_v2',
11
- # 'google_albert_xlarge_v2',
12
- # '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',
13
'google_electra_small',
14
- # 'google_electra_base',
15
- # 'google_electra_large',
16
- # 'google_uncased_mobilebert',
+ 'google_electra_base',
+ 'google_electra_large',
+ 'google_uncased_mobilebert',
17
'fairseq_bart_base',
18
- # 'fairseq_bart_large'
+ 'fairseq_bart_large'
19
]
20
21
# (batch_size, seq_length)
0 commit comments