1 files changed, 9 insertions, 9 deletions
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.actions.js b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
index 2c72f06..d067017 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.actions.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
@@ -202,18 +202,18 @@ 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({
-      type: types.app.load_progress,
-      progress: { i, n },
-      data: { module: 'pix2pixhd' },
-    })
+    // dispatch({
+    //   type: types.app.load_progress,
+    //   progress: { i, n },
+    //   data: { module: 'pix2pixhd' },
+    // })
   }).then(res => {
     const [sequence, datasetCount] = res //, datasets, results, output, datasetUsage, lossReport] = res
     const sequenceCount = sequence.length
@@ -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,