Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torch-scatter.git" did not exist on "3b9a9a900f933b1522d5d9a520c7e631cd138d73"
Unverified Commit 0f6f656f authored by Shining Sun's avatar Shining Sun Committed by GitHub
Browse files

bug fix (#6695) 

* bug fix

* bug fix
parent af47736d
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/resnet/keras/keras_benchmark.py
View file @ 0f6f656f
...@@ -37,7 +37,7 @@ class KerasBenchmark(tf.test.Benchmark): ...@@ -37,7 +37,7 @@ class KerasBenchmark(tf.test.Benchmark):
self.flag_methods = flag_methods or {} self.flag_methods = flag_methods or {}
if not output_dir: if not output_dir:
output_dir = '/tmp/' self.output_dir = '/tmp/'
def _get_model_dir(self, folder_name): def _get_model_dir(self, folder_name):
return os.path.join(self.output_dir, folder_name) return os.path.join(self.output_dir, folder_name)
......
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