From 0717df361a510e0ce02a4e8cb00796a4b897f02f Mon Sep 17 00:00:00 2001 From: Jules Laplace Date: Mon, 24 Sep 2018 00:36:21 +0200 Subject: check if dataset is generated --- app/client/modules/pix2pixhd/pix2pixhd.actions.js | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'app/client/modules/pix2pixhd/pix2pixhd.actions.js') diff --git a/app/client/modules/pix2pixhd/pix2pixhd.actions.js b/app/client/modules/pix2pixhd/pix2pixhd.actions.js index 44a42c4..d067017 100644 --- a/app/client/modules/pix2pixhd/pix2pixhd.actions.js +++ b/app/client/modules/pix2pixhd/pix2pixhd.actions.js @@ -202,11 +202,11 @@ export const list_epochs = (checkpoint_name) => (dispatch) => { }) } -export const count_dataset = (checkpoint_name) => (dispatch) => { +export const count_dataset = (dataset) => (dispatch) => { const module = pix2pixhdModule.name util.allProgress([ - actions.socket.list_directory({ module, dir: 'sequences/' + checkpoint_name + '/' }), - actions.socket.count_directory({ module, dir: 'datasets/' + checkpoint_name + '/train_A/' }), + actions.socket.list_directory({ module, dir: 'sequences/' + dataset + '/' }), + actions.socket.count_directory({ module, dir: 'datasets/' + dataset + '/train_A/' }), ], (percent, i, n) => { console.log('pix2pixhd load progress', i, n) // dispatch({ @@ -221,7 +221,7 @@ export const count_dataset = (checkpoint_name) => (dispatch) => { dispatch({ type: types.pix2pixhd.load_dataset_count, data: { - name: checkpoint_name, + name: dataset, sequence, sequenceCount, datasetCount, -- cgit v1.2.3-70-g09d2