Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tsoc
superbenchmark
Commits
e909ddd0
Unverified
Commit
e909ddd0
authored
Jun 16, 2023
by
guoshzhao
Committed by
GitHub
Jun 16, 2023
Browse files
Benchmarks - Update outdate references (#539)
**Description** Update 404 outdate reference links.
parent
f4dab9f7
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
8 deletions
+8
-8
superbench/benchmarks/model_benchmarks/pytorch_bert.py
superbench/benchmarks/model_benchmarks/pytorch_bert.py
+3
-3
superbench/benchmarks/model_benchmarks/pytorch_gpt2.py
superbench/benchmarks/model_benchmarks/pytorch_gpt2.py
+5
-5
No files found.
superbench/benchmarks/model_benchmarks/pytorch_bert.py
View file @
e909ddd0
...
...
@@ -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
,
...
...
superbench/benchmarks/model_benchmarks/pytorch_gpt2.py
View file @
e909ddd0
...
...
@@ -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'
)
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment