spacecruft
/
satnogs-wut
0
0
Fork 0
Code Issues 6 Releases Activity

update path

master
ml server 2020-01-21 13:20:35 -07:00
parent 7f1dcf4e62
commit 4aa809f80b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
wut-train-cluster-fn.py
View File

@ -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')