Unverified Commit e909ddd0 authored by guoshzhao's avatar guoshzhao Committed by GitHub
Browse files

Benchmarks - Update outdate references (#539)

**Description**
Update 404 outdate reference links.
parent f4dab9f7
......@@ -71,7 +71,7 @@ def __init__(self, name, parameters=''):
def add_parser_arguments(self):
"""Add the BERT-specified arguments.
BERT model reference: https://huggingface.co/transformers/model_doc/bert.html
BERT model reference: https://huggingface.co/docs/transformers/model_doc/bert
"""
super().add_parser_arguments()
......@@ -227,7 +227,7 @@ def _inference_step(self, precision):
# Register BERT Large benchmark.
# Reference: https://huggingface.co/transformers/pretrained_models.html
# Reference: https://huggingface.co/transformers/v3.3.1/pretrained_models.html
BenchmarkRegistry.register_benchmark(
'pytorch-bert-large',
PytorchBERT,
......@@ -235,7 +235,7 @@ def _inference_step(self, precision):
)
# Register BERT Base benchmark.
# Reference: https://huggingface.co/transformers/pretrained_models.html
# Reference: https://huggingface.co/transformers/v3.3.1/pretrained_models.html
BenchmarkRegistry.register_benchmark(
'pytorch-bert-base',
PytorchBERT,
......
......@@ -71,7 +71,7 @@ def __init__(self, name, parameters=''):
def add_parser_arguments(self):
"""Add the GPT2-specified arguments.
GPT2 model reference: https://huggingface.co/transformers/model_doc/gpt2.html
GPT2 model reference: https://huggingface.co/docs/transformers/model_doc/gpt2
"""
super().add_parser_arguments()
......@@ -221,25 +221,25 @@ def _inference_step(self, precision):
# Register GPT2 benchmark with 117M parameters.
# Reference: https://huggingface.co/transformers/pretrained_models.html
# Reference: https://huggingface.co/transformers/v3.3.1/pretrained_models.html
BenchmarkRegistry.register_benchmark(
'pytorch-gpt2-small', PytorchGPT2, parameters='--hidden_size=768 --num_hidden_layers=12 --num_attention_heads=12'
)
# Register GPT2 benchmark with 345M parameters.
# Reference: https://huggingface.co/transformers/pretrained_models.html
# Reference: https://huggingface.co/transformers/v3.3.1/pretrained_models.html
BenchmarkRegistry.register_benchmark(
'pytorch-gpt2-medium', PytorchGPT2, parameters='--hidden_size=1024 --num_hidden_layers=24 --num_attention_heads=16'
)
# Register GPT2 benchmark with 774M parameters.
# Reference: https://huggingface.co/transformers/pretrained_models.html
# Reference: https://huggingface.co/transformers/v3.3.1/pretrained_models.html
BenchmarkRegistry.register_benchmark(
'pytorch-gpt2-large', PytorchGPT2, parameters='--hidden_size=1280 --num_hidden_layers=36 --num_attention_heads=20'
)
# Register GPT2 benchmark with 1558M parameters.
# Reference: https://huggingface.co/transformers/pretrained_models.html
# Reference: https://huggingface.co/transformers/v3.3.1/pretrained_models.html
BenchmarkRegistry.register_benchmark(
'pytorch-gpt2-xl', PytorchGPT2, parameters='--hidden_size=1600 --num_hidden_layers=48 --num_attention_heads=25'
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment