Home
last modified time | relevance | path

Searched refs:model_names (Results 1 – 1 of 1) sorted by relevance

/test/mlts/benchmark/tools/
Dtensor_utils.py97 self.model_names = sorted(os.listdir(dump_dir))
135 def generate_animation_html(self, output_file_path, model_names=None, heatmap=True): argument
137 model_names = self.model_names if model_names is None else model_names
139 for model_name in tqdm(model_names):
169 model_names=None, heatmap=True): argument
174 model_names = self.model_names if model_names is None else model_names
178 for model_name in model_names:
188 for model_name in model_names:
542 generation_func(output_file_path=output_file_path, model_names=[args.model_name])