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

rename _cal_params_size as _cal_params_count. (#16)


Co-authored-by: default avatarGuoshuai Zhao <guzhao@microsoft.com>
parent 088aa19a
...@@ -336,7 +336,7 @@ class ModelBenchmark(Benchmark): ...@@ -336,7 +336,7 @@ class ModelBenchmark(Benchmark):
self._result.add_result(metric, avg) self._result.add_result(metric, avg)
@abstractmethod @abstractmethod
def _cal_params_size(self): def _cal_params_count(self):
"""Calculate the parameters scale of the model. """Calculate the parameters scale of the model.
Return: Return:
......
...@@ -149,7 +149,7 @@ class PytorchBase(ModelBenchmark): ...@@ -149,7 +149,7 @@ class PytorchBase(ModelBenchmark):
return True return True
def _cal_params_size(self): def _cal_params_count(self):
"""Calculate the parameters scale of the model. """Calculate the parameters scale of the model.
Return: Return:
......
...@@ -89,7 +89,7 @@ class FakeModelBenchmark(ModelBenchmark): ...@@ -89,7 +89,7 @@ class FakeModelBenchmark(ModelBenchmark):
duration.append(4) duration.append(4)
return duration return duration
def _cal_params_size(self): def _cal_params_count(self):
"""Calculate the parameters scale of the model. """Calculate the parameters scale of the model.
Return: Return:
......
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