Update benchmark_gluonnlp.py

dmlc · Aug 10, 2020 · 44f9c8b · 44f9c8b
1 parent 45c58b6
commit 44f9c8b
Showing 1 changed file with 10 additions and 10 deletions.
diff --git 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)