From 49e13753e7b69d08834c5893d987d573202a53f3 Mon Sep 17 00:00:00 2001 From: mkutsovsky Date: Wed, 5 Feb 2020 12:54:23 -0500 Subject: [PATCH] update to use data_dir for dropin fashion --- mnist.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mnist.py b/mnist.py index 877eb5f..925f725 100644 --- a/mnist.py +++ b/mnist.py @@ -98,7 +98,7 @@ def define_mnist_flags(): flags_core.define_base() flags_core.define_performance(num_parallel_calls=False) flags_core.define_image() - data_dir = os.path.abspath(os.environ.get('PS_JOBSPACE', os.getcwd()) + '/data') + data_dir = os.path.abspath(os.environ.get('DATA_DIR', os.getcwd() + '/data')) model_dir = os.path.abspath(os.environ.get('PS_MODEL_PATH', os.getcwd() + '/models') + '/mnist') export_dir = os.path.abspath(os.environ.get('PS_MODEL_PATH', os.getcwd() + '/models')) flags.adopt_module_key_flags(flags_core)