Add RoBERTa to the model suite (#2279)

Summary:
https://huggingface.co/FacebookAI/xlm-roberta-base

Pull Request resolved: #2279

Reviewed By: aaronenyeshi

Differential Revision: D59490274

Pulled By: xuzhao9

fbshipit-source-id: 1720f09e04603fc07c1825c2fb2f9062cda50701

torchbenchmark/models/hf_Roberta_base/__init__.py

@@ -0,0 +1,10 @@
+from torchbenchmark.tasks import NLP
+from torchbenchmark.util.framework.huggingface.model_factory import HuggingFaceModel
+
+class Model(HuggingFaceModel):
+    task = NLP.LANGUAGE_MODELING
+    DEFAULT_TRAIN_BSIZE = 4
+    DEFAULT_EVAL_BSIZE = 1
+
+    def __init__(self, test, device, batch_size=None, extra_args=[]):
+        super().__init__(name="hf_Roberta_base", test=test, device=device, batch_size=batch_size, extra_args=extra_args)

torchbenchmark/models/hf_Roberta_base/install.py

@@ -0,0 +1,14 @@
+
+import subprocess
+import sys
+import os
+from torchbenchmark.util.framework.huggingface.patch_hf import patch_transformers, cache_model
+
+def pip_install_requirements():
+    subprocess.check_call([sys.executable, '-m', 'pip', 'install', '-q', '-r', 'requirements.txt'])
+
+if __name__ == '__main__':
+    pip_install_requirements()
+    patch_transformers()
+    model_name = os.path.basename(os.path.dirname(os.path.abspath(__file__)))
+    cache_model(model_name)

torchbenchmark/models/hf_Roberta_base/metadata.yaml

@@ -0,0 +1,8 @@
+devices:
+  NVIDIA A100-SXM4-40GB:
+    eval_batch_size: 8
+eval_benchmark: false
+eval_deterministic: false
+eval_nograd: true
+train_benchmark: false
+train_deterministic: false

torchbenchmark/models/hf_Roberta_base/requirements.txt

@@ -0,0 +1,2 @@
+sentencepiece
+datasets

torchbenchmark/util/framework/huggingface/basic_configs.py

@@ -74,6 +74,12 @@
         "AutoModelForMaskedLM",
     ),
     "hf_Bert": (512, 512, "BertConfig()", "AutoModelForMaskedLM"),
+    "hf_Roberta_base": (
+        512,
+        512,
+        'AutoConfig.from_pretrained("xlm-roberta-base")',
+        "AutoModelForMaskedLM",
+    ),
     # see https://huggingface.co/bert-large-cased
     "hf_Bert_large": (
         512,