Added none check for output_dir (#6664)

* Added none check for output_dir * Change double quote to single

Added none check for output_dir (#6664)
* Added none check for output_dir * Change double quote to single
05b9122f · Shining Sun · Toby Boyd · 07678af8 · 05b9122f
Commit 05b9122f authored Apr 24, 2019 by Shining Sun Committed by Toby Boyd Apr 24, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

official/resnet/keras/keras_benchmark.py official/resnet/keras/keras_benchmark.py +3 -0

No files found.
--- a/official/resnet/keras/keras_benchmark.py
+++ b/official/resnet/keras/keras_benchmark.py
@@ -36,6 +36,9 @@ class KerasBenchmark(tf.test.Benchmark):
    self.default_flags = default_flags or {}
    self.flag_methods = flag_methods or {}

+    if not output_dir:
+      output_dir = '/tmp/'
+
  def _get_model_dir(self, folder_name):
    return os.path.join(self.output_dir, folder_name)