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
83a4e93f
Unverified
Commit
83a4e93f
authored
Mar 09, 2021
by
guoshzhao
Committed by
GitHub
Mar 09, 2021
Browse files
rename _cal_params_size as _cal_params_count. (#16)
Co-authored-by:
Guoshuai Zhao
<
guzhao@microsoft.com
>
parent
088aa19a
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
3 additions
and
3 deletions
+3
-3
superbench/benchmarks/model_benchmarks/model_base.py
superbench/benchmarks/model_benchmarks/model_base.py
+1
-1
superbench/benchmarks/model_benchmarks/pytorch_base.py
superbench/benchmarks/model_benchmarks/pytorch_base.py
+1
-1
tests/benchmarks/model_benchmarks/test_model_base.py
tests/benchmarks/model_benchmarks/test_model_base.py
+1
-1
No files found.
superbench/benchmarks/model_benchmarks/model_base.py
View file @
83a4e93f
...
...
@@ -336,7 +336,7 @@ def __process_model_result(self, model_action, precision, step_times):
self
.
_result
.
add_result
(
metric
,
avg
)
@
abstractmethod
def
_cal_params_
size
(
self
):
def
_cal_params_
count
(
self
):
"""Calculate the parameters scale of the model.
Return:
...
...
superbench/benchmarks/model_benchmarks/pytorch_base.py
View file @
83a4e93f
...
...
@@ -149,7 +149,7 @@ def _create_optimizer(self):
return
True
def
_cal_params_
size
(
self
):
def
_cal_params_
count
(
self
):
"""Calculate the parameters scale of the model.
Return:
...
...
tests/benchmarks/model_benchmarks/test_model_base.py
View file @
83a4e93f
...
...
@@ -89,7 +89,7 @@ def _inference_step(self, precision):
duration
.
append
(
4
)
return
duration
def
_cal_params_
size
(
self
):
def
_cal_params_
count
(
self
):
"""Calculate the parameters scale of the model.
Return:
...
...
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