Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "f282f6efd93ea1a8335caeb128b82fc4224112ad"
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
Hide 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