1 files changed, 19 insertions, 5 deletions
diff --git a/app/client/modules/pix2wav/pix2wav.actions.js b/app/client/modules/pix2wav/pix2wav.actions.js
index 9d819a3..48d0496 100644
--- a/app/client/modules/pix2wav/pix2wav.actions.js
+++ b/app/client/modules/pix2wav/pix2wav.actions.js
@@ -15,17 +15,19 @@ export const load_directories = (id) => (dispatch) => {
   const module = pix2wavModule.name
   util.allProgress([
     datasetLoader.load(module),
-    actions.socket.list_directory({ module, dir: 'sequences/pix2wav/' }),
-    actions.socket.list_directory({ module, dir: 'datasets/pix2wav/' }),
-    actions.socket.list_directory({ module, dir: 'checkpoints/pix2wav/' }),
+    actions.socket.list_directory({ module: 'pix2pix', dir: 'sequences/pix2wav/' }),
+    actions.socket.list_directory({ module: 'pix2pix', dir: 'datasets/pix2wav/' }),
+    actions.socket.list_directory({ module: 'pix2pix', dir: 'checkpoints/pix2wav/' }),
+    actions.socket.list_directory({ module: 'pix2pix', dir: 'checkpoints/pix2pix/' }),
     // actions.socket.list_directory({ module, dir: 'datasets' }),
     // actions.socket.list_directory({ module, dir: 'results' }),
     // actions.socket.list_directory({ module, dir: 'output' }),
     // actions.socket.disk_usage({ module, dir: 'datasets' }),
   ], (percent, i, n) => {
+    console.log('progress', i, n)
     dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
-    const [datasetApiReport, sequences, datasets, checkpoints] = res
+    const [datasetApiReport, sequences, datasets, checkpoints, pix2pixCheckpoints] = res
     const {
       folderLookup,
       fileLookup,
@@ -50,8 +52,20 @@ export const load_directories = (id) => (dispatch) => {
     checkpointDirectories.forEach(dir => {
       const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
       dataset.hasCheckpoints = true
+      dir.module = 'pix2wav'
     })
 
+    const pix2pixCheckpointDirectories = pix2pixCheckpoints.filter(s => s.dir)
+    pix2pixCheckpointDirectories.forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.hasCheckpoints = true
+      dir.module = 'pix2pix'
+    })
+
+    console.log(res)
+    console.log(checkpointDirectories)
+    console.log(pix2pixCheckpointDirectories)
+
     dispatch({
       type: types.dataset.load,
       data: {
@@ -62,7 +76,7 @@ export const load_directories = (id) => (dispatch) => {
         folders, files,
         sequences: sequenceDirectories,
         datasets,
-        checkpoints: checkpointDirectories,
+        checkpoints: checkpointDirectories.concat(pix2pixCheckpointDirectories),
       },
     })