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=''): ...@@ -71,7 +71,7 @@ def __init__(self, name, parameters=''):
def add_parser_arguments(self): def add_parser_arguments(self):
"""Add the BERT-specified arguments. """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() super().add_parser_arguments()
...@@ -227,7 +227,7 @@ def _inference_step(self, precision): ...@@ -227,7 +227,7 @@ def _inference_step(self, precision):
# Register BERT Large benchmark. # 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( BenchmarkRegistry.register_benchmark(
'pytorch-bert-large', 'pytorch-bert-large',
PytorchBERT, PytorchBERT,
...@@ -235,7 +235,7 @@ def _inference_step(self, precision): ...@@ -235,7 +235,7 @@ def _inference_step(self, precision):
) )
# Register BERT Base benchmark. # 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( BenchmarkRegistry.register_benchmark(
'pytorch-bert-base', 'pytorch-bert-base',
PytorchBERT, PytorchBERT,
......
...@@ -71,7 +71,7 @@ def __init__(self, name, parameters=''): ...@@ -71,7 +71,7 @@ def __init__(self, name, parameters=''):
def add_parser_arguments(self): def add_parser_arguments(self):
"""Add the GPT2-specified arguments. """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() super().add_parser_arguments()
...@@ -221,25 +221,25 @@ def _inference_step(self, precision): ...@@ -221,25 +221,25 @@ def _inference_step(self, precision):
# Register GPT2 benchmark with 117M parameters. # 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( BenchmarkRegistry.register_benchmark(
'pytorch-gpt2-small', PytorchGPT2, parameters='--hidden_size=768 --num_hidden_layers=12 --num_attention_heads=12' 'pytorch-gpt2-small', PytorchGPT2, parameters='--hidden_size=768 --num_hidden_layers=12 --num_attention_heads=12'
) )
# Register GPT2 benchmark with 345M parameters. # 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( BenchmarkRegistry.register_benchmark(
'pytorch-gpt2-medium', PytorchGPT2, parameters='--hidden_size=1024 --num_hidden_layers=24 --num_attention_heads=16' 'pytorch-gpt2-medium', PytorchGPT2, parameters='--hidden_size=1024 --num_hidden_layers=24 --num_attention_heads=16'
) )
# Register GPT2 benchmark with 774M parameters. # 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( BenchmarkRegistry.register_benchmark(
'pytorch-gpt2-large', PytorchGPT2, parameters='--hidden_size=1280 --num_hidden_layers=36 --num_attention_heads=20' 'pytorch-gpt2-large', PytorchGPT2, parameters='--hidden_size=1280 --num_hidden_layers=36 --num_attention_heads=20'
) )
# Register GPT2 benchmark with 1558M parameters. # 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( BenchmarkRegistry.register_benchmark(
'pytorch-gpt2-xl', PytorchGPT2, parameters='--hidden_size=1600 --num_hidden_layers=48 --num_attention_heads=25' '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