1 files changed, 4 insertions, 2 deletions
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.actions.js b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
index a17eeab..2c72f06 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.actions.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
@@ -205,7 +205,7 @@ export const list_epochs = (checkpoint_name) => (dispatch) => {
 export const count_dataset = (checkpoint_name) => (dispatch) => {
   const module = pix2pixhdModule.name
   util.allProgress([
-    actions.socket.count_directory({ module, dir: 'sequences/' + checkpoint_name + '/' }),
+    actions.socket.list_directory({ module, dir: 'sequences/' + checkpoint_name + '/' }),
     actions.socket.count_directory({ module, dir: 'datasets/' + checkpoint_name + '/train_A/' }),
   ], (percent, i, n) => {
     console.log('pix2pixhd load progress', i, n)
@@ -215,12 +215,14 @@ export const count_dataset = (checkpoint_name) => (dispatch) => {
       data: { module: 'pix2pixhd' },
     })
   }).then(res => {
-    const [sequenceCount, datasetCount] = res //, datasets, results, output, datasetUsage, lossReport] = res
+    const [sequence, datasetCount] = res //, datasets, results, output, datasetUsage, lossReport] = res
+    const sequenceCount = sequence.length
     console.log(sequenceCount, datasetCount)
     dispatch({
       type: types.pix2pixhd.load_dataset_count,
       data: {
         name: checkpoint_name,
+        sequence,
         sequenceCount,
         datasetCount,
       }