Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
__init__.py 337 Bytes
Newer Older
guoshzhao's avatar
Benchmarks: Initialization - Add base class, registry, and result (#1)  
guoshzhao committed
1
2
3
4
5 
# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

"""A module containing all the e2e model related benchmarks."""
guoshzhao's avatar
Benchmarks: Code Revision - Move benchmarks auto-registration from registry.py to __init__.py (#24)  
guoshzhao committed
6 
from superbench.benchmarks.model_benchmarks.model_base import ModelBenchmark
guoshzhao's avatar
Benchmarks: Add Benchmark - Add Pytorch BERT benchmarks, including bert-base...  
guoshzhao committed
7 
from superbench.benchmarks.model_benchmarks.pytorch_bert import PytorchBERT
guoshzhao's avatar
Benchmarks: Initialization - Add base class, registry, and result (#1)  
guoshzhao committed
8
guoshzhao's avatar
Benchmarks: Add Benchmark - Add Pytorch BERT benchmarks, including bert-base...  
guoshzhao committed
9 
__all__ = ['ModelBenchmark', 'PytorchBERT']