From 4aa809f80be2e0d4427352c036654a86b22aa957 Mon Sep 17 00:00:00 2001 From: ml server Date: Tue, 21 Jan 2020 13:20:35 -0700 Subject: [PATCH] update path --- wut-train-cluster-fn.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/wut-train-cluster-fn.py b/wut-train-cluster-fn.py index db37826..718ad0c 100644 --- a/wut-train-cluster-fn.py +++ b/wut-train-cluster-fn.py @@ -53,8 +53,9 @@ options = tf.data.Options() strategy = tf.distribute.experimental.MultiWorkerMirroredStrategy( tf.distribute.experimental.CollectiveCommunication.RING) -train_dir = os.path.join('data/', 'train') -val_dir = os.path.join('data/', 'val') +root_data_dir = ('/srv/satnogs') +train_dir = os.path.join(root_data_dir, 'data/', 'train') +val_dir = os.path.join(root_data_dir,'data/', 'val') train_good_dir = os.path.join(train_dir, 'good') train_bad_dir = os.path.join(train_dir, 'bad') val_good_dir = os.path.join(val_dir, 'good')