Skip to content

GitLab

"git@developer.sourcefind.cn:norm/vllm.git" did not exist on "c0c2335ce027486d254c31f665ce00d7db427d22"
Commit dd0ff4df authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 297381042
parent f2cf0cf4
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 4 deletions
official/benchmark/tfhub_memory_usage_benchmark.py
View file @ dd0ff4df
......@@ -30,10 +30,6 @@ FLAGS = flags.FLAGS
class TfHubMemoryUsageBenchmark(PerfZeroBenchmark):
"""A benchmark measuring memory usage for a given TF Hub SavedModel."""
def __init__(self, output_dir=None, default_flags=None, root_data_dir=None):
super(TfHubMemoryUsageBenchmark, self).__init__(
output_dir=output_dir, default_flags=default_flags)
def benchmark_memory_usage(self):
start_time_sec = time.time()
self.load_model()
......
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