Skip to content

GitLab

"docs/vscode:/vscode.git/clone" did not exist on "c46711e895bcec849fdcce69a7bc6864f023b6d1"
Commit 05b9122f authored by Shining Sun's avatar Shining Sun Committed by Toby Boyd
Browse files

Added none check for output_dir (#6664) 

* Added none check for output_dir

* Change double quote to single
parent 07678af8
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
official/resnet/keras/keras_benchmark.py
View file @ 05b9122f
...@@ -36,6 +36,9 @@ class KerasBenchmark(tf.test.Benchmark): ...@@ -36,6 +36,9 @@ class KerasBenchmark(tf.test.Benchmark):
self.default_flags = default_flags or {} self.default_flags = default_flags or {}
self.flag_methods = flag_methods or {} self.flag_methods = flag_methods or {}
if not output_dir:
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