Skip to content

Commit 44f9c8b

Browse files
committed
Update benchmark_gluonnlp.py
1 parent 45c58b6 commit 44f9c8b

1 file changed

Lines changed: 10 additions & 10 deletions

File tree

scripts/benchmarks/benchmark_gluonnlp.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -4,18 +4,18 @@
44

55

66
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',
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',
1313
'google_electra_small',
14-
# 'google_electra_base',
15-
# 'google_electra_large',
16-
# 'google_uncased_mobilebert',
14+
'google_electra_base',
15+
'google_electra_large',
16+
'google_uncased_mobilebert',
1717
'fairseq_bart_base',
18-
# 'fairseq_bart_large'
18+
'fairseq_bart_large'
1919
]
2020

2121
# (batch_size, seq_length)

0 commit comments

Comments
 (0)