Skip to content

GitLab

Commit dd0ff4df authored by A. Unique TensorFlower's avatar A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 297381042
parent f2cf0cf4
Show 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 ...@@ -30,10 +30,6 @@ FLAGS = flags.FLAGS
class TfHubMemoryUsageBenchmark(PerfZeroBenchmark): class TfHubMemoryUsageBenchmark(PerfZeroBenchmark):
"""A benchmark measuring memory usage for a given TF Hub SavedModel.""" """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): def benchmark_memory_usage(self):
start_time_sec = time.time() start_time_sec = time.time()
self.load_model() 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