1 files changed, 4 insertions, 4 deletions
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,